leetcode_1029 两地调度

要求

公司计划面试 2n 人。给你一个数组 costs ，其中 costs[i] = [aCosti, bCosti] 。第 i 人飞往 a 市的费用为 aCosti ，飞往 b 市的费用为 bCosti 。

返回将每个人都飞到 a 、b 中某座城市的最低费用，要求每个城市都有 n 人抵达。

示例 1：

输入：costs = [[10,20],[30,200],[400,50],[30,20]]
输出：110
解释：
第一个人去 a 市，费用为 10。
第二个人去 a 市，费用为 30。
第三个人去 b 市，费用为 50。
第四个人去 b 市，费用为 20。

最低总费用为 10 + 30 + 50 + 20 = 110，每个城市都有一半的人在面试。

示例 2：

输入：costs = [[259,770],[448,54],[926,667],[184,139],[840,118],[577,469]]
输出：1859
示例 3：

输入：costs = [[515,563],[451,713],[537,709],[343,819],[855,779],[457,60],[650,359],[631,42]]
输出：3086

提示：

• 2 * n == costs.length
• 2 <= costs.length <= 100
• costs.length 为偶数
• 1 <= aCosti, bCosti <= 1000

核心代码

class Solution:
    def twoCitySchedCost(self, costs: List[List[int]]) -> int:
        costs = sorted(costs,key = lambda x:abs(x[0] - x[1]))
        costs = costs[::-1]
        a,b = 0,0
        n = len(costs) // 2
        res = 0

        for i,cost in enumerate(costs):
            if a < n and b < n:
                if cost[0] < cost[1]:
                    a += 1
                    res += cost[0]
                else:
                    b += 1
                    res += cost[1]
            elif a < n:
                a += 1
                res += cost[0]
            elif b < n:
                b += 1
                res += cost[1]
        return res

解题思路：贪心算法，先把输入数组按照两地费用之差的绝对值排好序(从大到小，说明差值越大，去小的越划算)，从前到后线性扫描，然后在去A市和去B市的名额都还有空余的时候，每一组费用取较小的那个，如果去A市的名额已经满了，剩下的全部去B，同理如果去B市的已经满了，剩下的全部去A。