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;
    }
};

 

发布了379 篇原创文章 · 获赞 105 · 访问量 4万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 精致技术 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览