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]

方法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

你可能感兴趣的:(LeetCode-python 1029.两地调度)