LeetCode 1029. Two City Scheduling

链接：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;
    }
};