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

提示：

1. 1 <= costs.length <= 100
2. costs.length 为偶数
3. 1 <= costs[i][0], costs[i][1] <= 1000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-city-scheduling
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：先求costs[i][0] - costs[i][1]的差值，对差值排序，同时交换 costs[i]与costs[j]， 差值的结果从小到大排序。前N个数，选择costs[i][0],即去A城市， 后N个数，选择costs[i][1]，即去B城市。

举例子：其实每个人去哪个城市，主要取决于ta去两个城市的差值。1号人物：[100,110]， 2号人物：[10, 50]。 1号人物的差值为 -10， 2号人物的差值为 -40. 所以2号人物先选择A城市，1号人物，去B城市。总的花费最少。

class Solution {
public:
    int twoCitySchedCost(vector<vector<int>>& costs) {
        vector<int> chazhi;
        // 1. 先计算差值
        for(int i = 0; i < costs.size(); ++i)
        {
            chazhi.push_back(costs[i][0] - costs[i][1]);
        }
        // 2. 根据差值排序，同时改变每个人去城市花费的顺序
        for(int i = 0; i < costs.size()-1; ++i)
        {
            bool flag = true;
            for(int j = 0; j < costs.size()-i-1; ++j)
            {
                if(chazhi[j] > chazhi[j+1])
                {
                    swap(chazhi[j], chazhi[j+1]);
                    swap(costs[j][0], costs[j+1][0]);
                    swap(costs[j][1], costs[j+1][1]);
                    flag = false;
                }
            }
            if (flag)
		    {
			    break;
		    }
        }
        // 前一半去A城市
        int total = 0;
        for(int i = 0; i < costs.size()/2; ++i)
        {
            total += costs[i][0];
        }
        // 后一半去B城市
        for(int i = costs.size()/2; i < costs.size(); ++i)
        {
            total += costs[i][1];
        }
        return total;
    }
};