跳转至

LeetCode: 1029. 两地调度

1、题目描述

公司计划面试 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$

2、解题思路

  • 按照去A和去B哪个更更小排序
  • 前面一半表示去A更小

如何判断一个人去A还是去B呢?

如果去A的代价更小,就去A,按照代价排序,代价最小的一半去A

class Solution:
    def twoCitySchedCost(self, costs: List[List[int]]) -> int:
        sums = 0

        nums = sorted(costs, key=lambda x: x[0] - x[1])

        for index, num in enumerate(nums):
            sums += nums[index][index // (len(costs) // 2)]

        return sums