Leetcode—priorityQueue(1)

373. Find K Pairs with Smallest Sums ✖️

class Solution {
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        PriorityQueue<int[]> pq = new PriorityQueue<>((a, b) -> (a[0] + a[1]) - (b[0] + b[1]));
// PriorityQueue<int[]> 是一个优先队列，其元素为整数数组 (int[])。这里使用了一个 lambda 表达式 (a, b) -> (a[0] + a[1]) - (b[0] + b[1]) 作为比较器。在这个 lambda 表达式中：

// a 和 b 都是队列中的元素，这里的元素是整数数组类型 (int[])。
// a[0] + a[1] 和 b[0] + b[1] 分别是数组 a 和 b 的前两个元素之和。
        for(int i = 0; i < nums1.length && i < k; i++) {
            pq.add(new int[]{nums1[i], nums2[0], 0});
        }
        List<List<Integer>> res = new ArrayList();
        for(int i = 0; i < k && !pq.isEmpty(); i++) {
            int cur[] = pq.poll();
            res.add(List.of(cur[0], cur[1]));
            int nums2Idx = cur[2];
            if(nums2Idx < nums2.length - 1) {
                pq.add(new int[]{cur[0], nums2[nums2Idx + 1], nums2Idx + 1});
            }
        }
        return res;
    }
}