1029. 两地调度

链接：https://leetcode-cn.com/problems/two-city-scheduling/

公司计划面试 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

思路：我们把这个问题看作先所有人都去B市，那么有一半人需要再去A市，那么如何选择这一半人，我们通过costs[i][0] - costs[i][1]的大小来筛选，有正有负，负的说明去A市的花费更少，因为我们之前已经假定所有人都到了B市，那么再去A市当然是花费越小的越好。

比如[10,20],[30,200],虽前者去A、B市的花费都小于后者，但后者去B市与去A市的花费差距大于前者，因此应该后者去A市。

我们用diff记录每个人的costs[i][0] - costs[i][1]，然后从小到大排序，计算总花费时只需要所有到B市的花费再加上diff前N个的花费

class Solution {
public:
    int twoCitySchedCost(vector<vector<int>>& costs) {
        int ans = 0;
        vector<int> diff;
        for(int i = 0; i < costs.size(); i++){
            diff.push_back(costs[i][0] - costs[i][1]);
        }

        sort(diff.begin(), diff.end());
        
        for(int i = 0; i < diff.size(); i++){
            ans = ans + costs[i][1];
            if(i < diff.size() / 2){
                ans = ans + diff[i];
            }
        }

        return ans;
    }
};