两地调度

/*
 * 思路：
 * 1. 计算每个人去a市和b市的成本差
 * 2. 根据成本差对costs进行排序
 * 3. 选择前n个人去a市，后n个人去b市
 */

import java.util.Arrays;
import java.util.Comparator;

public class Solution {
    public int twoCitySchedCost(int[][] costs) {
        return Arrays.stream(costs)
                // 按照去a市和b市的成本差进行排序
                .sorted(Comparator.comparingInt(a -> a[0] - a[1]))
                // 将前n个人去a市的费用和后n个人去b市的费用相加
                .mapToInt((cost, index) -> index < costs.length / 2 ? cost[0] : cost[1])
                .sum();
    }

    public static void main(String[] args) {
        Solution sol = new Solution();
        int[][] costs = {{10, 20}, {30, 200}, {400, 50}, {30, 20}};
        System.out.println(sol.twoCitySchedCost(costs));  // 输出: 110
    }
}