leecode1029. 两地调度

公司计划面试 2n 人。给你一个数组 costs ,其中 costs[i] = [aCosti, bCosti] 。第 i 人飞往 a 市的费用为 aCosti ,飞往 b 市的费用为 bCosti 。

返回将每个人都飞到 a 、b 中某座城市的最低费用,要求每个城市都有 n 人抵达。

示例 1:

输入:costs = [[10,20],[30,200],[400,50],[30,20]]
输出:110
解释:
第一个人去 a 市,费用为 10。
第二个人去 a 市,费用为 30。
第三个人去 b 市,费用为 50。
第四个人去 b 市,费用为 20。

最低总费用为 10 + 30 + 50 + 20 = 110,每个城市都有一半的人在面试。
示例 2:

输入:costs = [[259,770],[448,54],[926,667],[184,139],[840,118],[577,469]]
输出:1859
示例 3:

输入:costs = [[515,563],[451,713],[537,709],[343,819],[855,779],[457,60],[650,359],[631,42]]
输出:3086

使用记忆化搜索:

class Solution {
    public int twoCitySchedCost(int[][] costs) {
        //截止到第几个人去a,b市分别要的最小费用
        int[][][] map = new int[costs.length][costs.length/2 + 1][costs.length/2 + 1];
        return dfs(costs, 0, costs.length / 2, costs.length / 2, map);
    }

    public int dfs(int[][] costs, int index, int aRest, int bRest, int[][][] map) {
        if (index == costs.length) {
            return 0;
        }
        if (map[index][aRest][bRest] != 0) {
            return map[index][aRest][bRest];
        }
        //当前人去A,B市的费用数组
        int[] cost = costs[index];
        if (aRest >= 0 && bRest >= 0) {
            int r1 = 99999999;
            int r2 = 99999999;
            //可以去A市
            if (aRest != 0) {
                r1 = cost[0] + dfs(costs, index + 1, aRest - 1, bRest, map);
            }
            //可以去B市
            if (bRest != 0) {
                r2 = cost[1] + dfs(costs, index + 1, aRest, bRest - 1, map);
            }
            map[index][aRest][bRest] = Math.min(r1, r2);
            return map[index][aRest][bRest];
        }
        return 0;
    }

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值