题目描述
公司计划面试 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
算法1
(排序) $O(nlogn)$
这道题先假设所有的人都去A市,然后我们只需要选取去B市的费用与去A市的费用的差最小的N个人改去B市即可。
时间复杂度分析:排序时间复杂度$O(nlogn)$
Python3 代码
class Solution:
def twoCitySchedCost(self, costs: List[List[int]]) -> int:
dif = [cost[1]-cost[0] for cost in costs]
res = sum([cost[0] for cost in costs])
dif.sort()
for i in range(len(costs)//2):
res += dif[i]
return res