查找和最小的 K 对数字

/*
 * 思路：
 * 1. 使用嵌套的流生成所有可能的数对，并按和排序。
 * 2. 取出前 k 个数对。
 */
import java.util.*;
import java.util.stream.*;

public class Solution {
    public List<int[]> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        return Arrays.stream(nums1)
            .boxed()
            .flatMap(u -> Arrays.stream(nums2).mapToObj(v -> new int[]{u, v}))
            .sorted(Comparator.comparingInt(a -> a[0] + a[1]))
            .limit(k)
            .collect(Collectors.toList());
    }
}