题目来源 LeetCode

算法标签 贪心

题目描述

公司计划面试 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
通过次数9,070提交次数14,866

思路

贪心

题目的意思就是两个城市都有一半人
要最优惠的价格

则 我们假设所有人去A,我们则肯定需要一半人去B
怎么选择着一半人,则是查找B-A正数差值最大的一半人

减去差值之后我们就相当于一半人去了A,一半人去了B,且是最优惠价格

AC代码

class Solution {
public:
int twoCitySchedCost(vector<vector<int>>& costs) {
int sum = 0;
for(int i=0;i<costs.size();i++)sum+=costs[i][0];

vector<int> arr;
for(int i=0;i<costs.size();i++)arr.push_back(costs[i][0]-costs[i][1]);

sort(arr.begin(), arr.end());

for(int i=arr.size()-1;i>=arr.size()>>1;i--)sum-=arr[i];

return sum;
}
};