【Leetcode】1029. Two City Scheduling

题目地址：

https://leetcode.com/problems/two-city-scheduling/

给定$2n$个面试者，给定他们分别去城市$A$和城市$B$的花费，以数组形式表示，记为$C$。公司打算让其中$n$个人去城市$A$面试，让剩下的$n$个人去城市$B$面试。问怎样安排使得总花费最小。返回最小花费。

思路是贪心。假设先让所有人都去城市$A$面试，那么总花费就是${\sum }_{i=0}^{2n-1}C[i][0]$。现在考虑将其中$n$个人分配到城市$B$中去，显然应该尽可能将分配到$B$之后花费会变小的人分配过去。对于下标为$i$的人来说，将他分配到$B$之后，总花费会减少$C[i][1]-C[i][0]$（这里的意思是净减少，有可能$C[i][1]-C[i][0]<0$，这样分配过后总花费实际上是增加的）。那么就应该将$C[i][1]-C[i][0]$尽可能大的那些人分到$B$那里去，其余分到$A$那里。所以思路就是，将数组$C$按照$C[i][1]-C[i][0]$排序，前一半的人分到$A$，其余分到$B$即可。代码如下：

class Solution {
 public:
  int twoCitySchedCost(vector<vector<int>> &costs) {
    sort(costs.begin(), costs.end(),
         [&](auto &v1, auto &v2) { return v1[0] - v1[1] < v2[0] - v2[1]; });
    int res = 0, n = costs.size();
    for (int i = 0; i < n / 2; i++) res += costs[i][0];
    for (int i = n / 2; i < n; i++) res += costs[i][1];
    return res;
  }
};

时间复杂度$O(n\log n)$，空间$O(1)$。