Leetcode 1029. 两地调度

Leetcode 1029. 两地调度

题目

公司计划面试 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，每个城市都有一半的人在面试。

思路

• 假设让这些人全部都去A， 问题就变成了从中选择一半人让他们去B
• 假设甲去A —— 50， 甲去B —— 10， 乙去A —— 50， 乙去B —— 30， 那么如果让甲换成去B， 可以节省40， 如果让乙换成去B， 可以节省20， 那么当然选择让甲换成去B了
• 所以先统计所有去A的费用， 再用堆 —— 优先级队列去选择一半人出来去B

注：
如果是全局的排序， 那么可以sort
像这样的只选一半的部分排序， 可以用堆解决

代码

class Solution {
public:
    int twoCitySchedCost(vector<vector<int>>& costs) {
        int res = 0, size = costs.size();
        priority_queue<int> q;

        for (int i = 0;i < size;++i) {
            res += costs[i][0];
            q.push(costs[i][0] - costs[i][1]);
        }

        for (int i = 0;i < size / 2;++i) {
            res -= q.top();
            q.pop();
        }

        return res;
    }
};