题目描述
公司计划面试 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]);
}
};