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 <= costs.length <= 100
costs.length 为偶数
1 <= costs[i][0], costs[i][1] <= 100

方法1:
主要思路:解题汇总链接
(1)贪心;
(2)将原数组的元素根据到两个城市的成本的差值的绝对值进行排序,则从头到尾遍历排序后的数组,到那个城市的成本低,就到那个城市,知道到达某个城市的人数达到了限制的一半,则将剩下人都安排到去另一个城市;

class Solution {
public:
    int twoCitySchedCost(vector<vector<int>>& costs) {
        sort(costs.begin(),costs.end(),[](vector<int>&lhs,vector<int>&rhs){//根据到两个城市的差值进行排序
            return abs(lhs[0]-lhs[1])>abs(rhs[0]-rhs[1]);
        });
        int count_A=0,count_B=0;//统计到两个城市的人数
        int sum_A=0,sum_B=0;//统计到两个城市的成本
        int index=0;//遍历数组的索引
        while(index<costs.size()){
            if(costs[index][0]<costs[index][1]){//选择成本较小的城市
                ++count_A;
                sum_A+=costs[index][0];
            }
            else{
                ++count_B;
                sum_B+=costs[index][1];
            }
            ++index;
            if(count_A*2==costs.size()||count_B*2==costs.size()){//说明已经有某个城市的人数达到要求
                break;
            }
        }
        //将剩余的人安排到另一个城市
        if(count_A*2==costs.size()){
            while(index<costs.size()){
                ++count_B;
                sum_B+=costs[index][1];
                ++index;
            }
        }
        else{
            while(index<costs.size()){
                ++count_A;
                sum_A+=costs[index][0];
                ++index;
            }
        }
        return sum_A+sum_B;//返回总的成本
    }
};

方法2:
主要思路:
(1)同样是贪心,只不过排序的函数不同,直接根据差异,将其分成前后两组进行操作;

class Solution {
public:
    int twoCitySchedCost(vector<vector<int>>& costs) {
        sort(costs.begin(),costs.end(),[](vector<int>&lhs,vector<int>&rhs){
            return lhs[0]-lhs[1]>rhs[0]-rhs[1];
        });
        int sum_A=0,sum_B=0;
        int end=costs.size()/2;
        for(int i=0;i<end;++i){
            sum_A+=costs[i][1];
            sum_B+=costs[i+end][0];
        }
        return sum_A+sum_B;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值