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,每个城市都有一半的人在面试。
提示:
1 <= costs.length <= 100
costs.length 为偶数
1 <= costs[i][0], costs[i][1] <= 1000
通过次数5,982提交次数10,249

思路:
总人数一定是偶数,一半分到A市,一半到B市,每个人的选择非A即B。
判断每个人去A还是B:选择cost较低的
cost[i][0]>cost[i][1] : A cost[i][0]<cost[i][1] : B
但是这样未必保证到A、B人数相同啊,所以不能就这样确定去A还是B

总结思路:考虑员工去AB的差值,对差值进行排列,假设全都去A,再挑选其中一半变成B,利用AB差值排序,排序的前N个就选择B,
总代价=A代价求和+AB差值前N个
(只是计算代价也不需要知道哪个员工去A还是B)
Python代码如下:

class Solution(object):
    def twoCitySchedCost(self, costs):
        """
        :type costs: List[List[int]]
        :rtype: int
        """
        res = 0
        diff = []
        for i in range(len(costs)):
            res = res + costs[i][0]
            a = costs[i][1]-costs[i][0]
            diff.append(a)
        diff.sort()
        del diff[len(diff)/2:len(diff)]
        return res+sum(diff)

参考一下题解:对cost列表中元素按照 A-B的差值从小到大排序,(从序列尾观察就是B-A的差值从小到大排序)则前N个去A地,后N个去B地

class Solution:
    def twoCitySchedCost(self, costs):
        costs.sort(key=lambda x: (x[0]-x[1]))  # 计算去A地和去B低的费用差,然后按照费用差排序
        length_costs = len(costs)
        result = 0
        result += sum([i[0] for i in costs[:length_costs//2]])  # 前半部分去A地
        result += sum([i[1] for i in costs[length_costs//2:]])  # 后半部分去B地
        return result
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值