[LeetCode] 1029、两地调度

题目描述

公司计划面试 2N 人。第 i 人飞往 A 市的费用为 costs[i][0],飞往 B 市的费用为 costs[i][1]。返回将每个人都飞到某座城市的最低费用,要求每个城市都有 N 人抵达。

输入:[[10,20],[30,200],[400,50],[30,20]]
输出:110
解释:
第一个人去 A 市,费用为 10。
第二个人去 A 市,费用为 30。
第三个人去 B 市,费用为 50。
第四个人去 B 市,费用为 20。
最低总费用为 10 + 30 + 50 + 20 = 110,每个城市都有一半的人在面试。

解题思路

最优的方案是,根据price_A - price_B的大小进行排序,然后前一半去A地,后一半去B地。

参考代码

class Solution {
public:
    int twoCitySchedCost(vector<vector<int> >& costs) {
        int length = costs.size();
        sort(costs.begin(), costs.end(), cmp);
        
        int res = 0;
        for(int i = 0; i < length/2; i++)
            res += costs[i][0];
        for(int i = length/2; i < length; i++)
            res += costs[i][1];
        
        return res;
    }
    
    static bool cmp(vector<int> a, vector<int> b){
        return (a[0] - a[1]) < (b[0] - b[1]);
    }
    
};
发布了390 篇原创文章 · 获赞 576 · 访问量 15万+

猜你喜欢

转载自blog.csdn.net/ft_sunshine/article/details/103945236