一、题干
公司计划面试 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
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/two-city-scheduling
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、思路
先考虑所有人都去A地,然后派n个人去A地,派出A地的情况就是,A的路费减去B的路费,我们需要选择这个值最小的人去A地。
我们可以这么做,按照A-B的路费的大小递增排序,前n个派往A地,其余的派往B地。
这样就可以使得最后的路费是最低的,这也是一种贪心思想
三、代码实现
class Solution {
public:
int twoCitySchedCost(vector<vector<int>>& costs) {
sort(costs.begin(), costs.end(),[&](const vector<int>& a, const vector<int>& b) {
return a[0] - a[1] < b[0] - b[1];
});//先按照A-B的大小进行排序
int n = costs.size()/2;
int ret = 0;
for (int i = 0;i < n;i++) {//然后把费用加起来
ret += costs[i][0] + costs[i + n][1];
}
return ret;
}
};
版权声明:本文为qq_42331828原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。