链接:https://leetcode.com/problems/two-city-scheduling/
思路
贪心法。2N个人选N个人去B,则B-A最小的那个肯定去B。证明:设最小的那个人p1,cost分别为A1 B1。在某种分法中,如果它没有去B,选择一个去了B的人p2,设cost为A2 B2。将这两人交换,即改成p1去B,p2去A,则总cost增加-A1+B1-B2+A2=(B1-A1)-(B2-A2),为负数。说明无论如何分配,如果p1去了A,那一定能找到另一种分法,使p1去B且cost更小,即p1必然去B。确定了p1去B,剩下的问题就是2N-1个人选N-1个人去B,依此类推,最后变成N个人选0个人去B,不需要再处理。总之,找出2N个里,B-A最小的N个让他们去B即可。
代码
class Solution {
public:
int twoCitySchedCost(vector<vector<int>>& costs) {
int n = costs.size(), N = n / 2;
vector<pair<int, int>> diffs(n); // B-A, idx
for(int i = 0; i < n; i++) {
diffs[i] = {costs[i][1] - costs[i][0], i};
}
sort(diffs.begin(), diffs.end());
int sum = 0;
// B-A小的选B
for(int i = 0; i < N; i++) {
sum += costs[diffs[i].second][1];
}
for(int i = N; i < n; i++) {
sum += costs[diffs[i].second][0];
}
return sum;
}
};