【leetcode刷题】T182-两地调度

示例：
输入：[[10,20],[30,200],[400,50],[30,20]]
输出：110
解释：
第一个人去 A 市，费用为 10。
第二个人去 A 市，费用为 30。
第三个人去 B 市，费用为 50。
第四个人去 B 市，费用为 20。
最低总费用为 10 + 30 + 50 + 20 = 110，每个城市都有一半的人在面试。

class Solution(object):
    def twoCitySchedCost(self, costs):
        """
        :type costs: List[List[int]]
        :rtype: int
        """
        res = sum(map(lambda x: x[0], costs))
        res += sum(sorted(map(lambda x: x[1] - x[0], costs))[:len(costs) // 2])
        return res

class Solution {
public:
    int twoCitySchedCost(vector<vector<int>>& costs) {
        int res = 0;
        vector<int> tmp;
        for(int i=0; i<costs.size(); i++){
            res += costs[i][0];
            tmp.push_back(costs[i][1] - costs[i][0]);
        }
        sort(tmp.begin(), tmp.end());
        for(int i=0; i<costs.size()/2; i++)
            res += tmp[i];
        return res;
    }
};