Leetcode 1029. 两地调度

题目描述

公司计划面试 2N 人。第 i 人飞往 A 市的费用为 costs[i][0]，飞往 B 市的费用为 costs[i][1]。

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

示例 1:

输入：[[10,20],[30,200],[400,50],[30,20]]

输出：110

解释：

第一个人去 A 市，费用为 10。
第二个人去 A 市，费用为 30。
第三个人去 B 市，费用为 50。
第四个人去 B 市，费用为 20。

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

解法

以 sub 数组表示去 B 市与去 A 市的代价差值，即 sub[i] = cost[i][1] - cost[i][0]，对 sub 递减排序。不妨以 ret 表示 2N 人全部去了 B 市的总费用，则 ret - sum(sub[:N]) 表示有 N 人去了 A 市的最低总费用。

class Solution:
    def twoCitySchedCost(self, costs: List[List[int]]) -> int:
        ret,sub=0,[]
        for i in costs:
            ret+=i[1]
            sub.append(i[1]-i[0])
        sub.sort(reverse=True)
        return ret-sum(sub[:len(costs)//2])

题中为一半的人去了 A 市，该方式可同样计算出总人数的三分之一、四分之一人数去 A 市的最小费用问题。