Python Leetcode(1029.两地调度)

Python Leetcode(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 <= costs.length <= 100
costs.length 为偶数
1 <= costs[i][0], costs[i][1] <= 1000

Solution:(要点在于分配哪部分人去A市，哪部分人去B市。利用A市费用减去B市费用，这个值越小说明派这个人去A市越划算。所以把每个人的差值计算出来，进行排序，前N个人就派去A市，后N个人派去B市。)

class Solution(object):
    def twoCitySchedCost(self, costs):
        """
        :type costs: List[List[int]]
        :rtype: int
        """
        count_minors = []
        l, result = len(costs), 0
        
        # 在数组中添加差值和去A市的费用
        for each in costs:
            count_minors.append([each[0], each[0] - each[1]])
        
        # 利用差值进行排序
        count_minors = sorted(count_minors, key=lambda x: x[1])
        
        for i in range(int(l/2)):
            result += count_minors[i][0]
        for j in range(int(l/2), l):
            result += count_minors[j][0] - count_minors[j][1]
        
        return result

solution = Solution()
print(solution.twoCitySchedCost([[10,20],[30,200],[400,50],[30,20]]))

110