LeetCode-python 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:
两地费用差的绝对值大的先安排

方法2:
两地费用求差,从小到大排序,前一半去a, 后一半去b

代码实现

方法1:

class Solution:
    def twoCitySchedCost(self, costs: List[List[int]]) -> int:
        costs = sorted(costs, key = lambda x: abs(x[0]-x[1]), reverse=True)    
        n = len(costs)//2  
        i, j = 0, 0
        res = 0
        for cost in costs:
            if cost[0]<cost[1] and i<n :
                res += cost[0]
                i += 1
            elif j<n:
                res += cost[1]
                j += 1
            else:
                res += cost[0]
        return res

方法2:

class Solution:
    def twoCitySchedCost(self, costs: List[List[int]]) -> int:
        costs.sort(key=lambda x : (x[0]-x[1]))         
        n = len(costs)//2
        return sum([x[0] for x in costs[:n]]) + sum([x[1] for x in costs[n:]])

本文链接:https://www.jianshu.com/p/aecc0931f672

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 专业考题类型管理运行工作负责人一般作业考题内容选项A选项B选项C选项D选项E选项F正确答案 变电单选GYSZ本规程...
    小白兔去钓鱼阅读 13,402评论 0 13
  • 题目描述 公司计划面试 2N 人。第 i 人飞往 A 市的费用为 costs[i][0],飞往 B 市的费用为 c...
    zhipingChen阅读 2,695评论 0 2
  • 更多精彩内容,请关注【力扣简单题】。 题目 难度:★★☆☆☆类型:数学 公司计划面试 2N 人。第 i 人飞往 A...
    玖月晴阅读 3,273评论 0 0
  • 这是小川的第383次更新,第412篇原创 01 看题和准备 今天介绍的是LeetCode算法题中Easy级别的第2...
    程序员小川阅读 3,870评论 0 2
  • 三月,木棉花开,春风染指。 写不尽的红,道不完的暖。 路过,郁郁森林,缄口不言。 如何说爱你,如这般风和日丽。 你...
    一言尔阅读 1,351评论 3 3

友情链接更多精彩内容