【题目】
公司计划面试 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,每个城市都有一半的人在面试。
提示: 1 <= costs.length <= 100 costs.length 为偶数 1 <= costs[i][0], costs[i][1] <= 1000
【思路】
每个人都可以选择任意一个目的地,那么可以都先选择A市,求得总费用,再更改一部分人的目的地,即计算所有的额外费用=B市费用-A市费用,排序并且选择最小的N个,将额外费用加入到总费用中。
【代码】
python版本
class Solution(object):
def twoCitySchedCost(self, costs):
"""
:type costs: List[List[int]]
:rtype: int
"""
res = sum(map(lambda x: x[0], costs))
res += sum(sorted(map(lambda x: x[1] - x[0], costs))[:len(costs) // 2])
return res
C++版本
class Solution {
public:
int twoCitySchedCost(vector<vector<int>>& costs) {
int res = 0;
vector<int> tmp;
for(int i=0; i<costs.size(); i++){
res += costs[i][0];
tmp.push_back(costs[i][1] - costs[i][0]);
}
sort(tmp.begin(), tmp.end());
for(int i=0; i<costs.size()/2; i++)
res += tmp[i];
return res;
}
};