Leetcode1029. 两地调度

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

tips: 衡量某个人去某个城市的变量是该人去两城市需要的代价之差,差越大,越倾向于去代价小的那个。那么如何取一半的人去某个城市,需要对所有人的两城市之差做个排序。假设所有人去B城市,此时花费为res,按照cost[1]-cost[0]从大至小排序,前N个人去往A城市,那么最终的花费为 res-=sum(sorted(sub[:N])),代码如下:

class Solution:
    def twoCitySchedCost(self, costs: List[List[int]]) -> int:
        res,sub=0,[]
        for c in costs:
            res+=c[1]
            sub.append(c[1]-c[0])
        sub.sort(reverse=1)
        res-=sum(sub[:len(costs)//2])
        return res

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值