题目描述:
公司计划面试 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] <= 100
方法1:
主要思路:解题汇总链接
(1)贪心;
(2)将原数组的元素根据到两个城市的成本的差值的绝对值进行排序,则从头到尾遍历排序后的数组,到那个城市的成本低,就到那个城市,知道到达某个城市的人数达到了限制的一半,则将剩下人都安排到去另一个城市;
class Solution {
public:
int twoCitySchedCost(vector<vector<int>>& costs) {
sort(costs.begin(),costs.end(),[](vector<int>&lhs,vector<int>&rhs){//根据到两个城市的差值进行排序
return abs(lhs[0]-lhs[1])>abs(rhs[0]-rhs[1]);
});
int count_A=0,count_B=0;//统计到两个城市的人数
int sum_A=0,sum_B=0;//统计到两个城市的成本
int index=0;//遍历数组的索引
while(index<costs.size()){
if(costs[index][0]<costs[index][1]){//选择成本较小的城市
++count_A;
sum_A+=costs[index][0];
}
else{
++count_B;
sum_B+=costs[index][1];
}
++index;
if(count_A*2==costs.size()||count_B*2==costs.size()){//说明已经有某个城市的人数达到要求
break;
}
}
//将剩余的人安排到另一个城市
if(count_A*2==costs.size()){
while(index<costs.size()){
++count_B;
sum_B+=costs[index][1];
++index;
}
}
else{
while(index<costs.size()){
++count_A;
sum_A+=costs[index][0];
++index;
}
}
return sum_A+sum_B;//返回总的成本
}
};
方法2:
主要思路:
(1)同样是贪心,只不过排序的函数不同,直接根据差异,将其分成前后两组进行操作;
class Solution {
public:
int twoCitySchedCost(vector<vector<int>>& costs) {
sort(costs.begin(),costs.end(),[](vector<int>&lhs,vector<int>&rhs){
return lhs[0]-lhs[1]>rhs[0]-rhs[1];
});
int sum_A=0,sum_B=0;
int end=costs.size()/2;
for(int i=0;i<end;++i){
sum_A+=costs[i][1];
sum_B+=costs[i+end][0];
}
return sum_A+sum_B;
}
};