373. 查找和最小的K对数字

给定两个以升序排列的整数数组 nums1 和 nums2 , 以及一个整数 k 。

定义一对值 (u,v)，其中第一个元素来自 nums1，第二个元素来自 nums2 。

请找到和最小的 k 个数对 (u1,v1),  (u2,v2)  ...  (uk,vk) 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-k-pairs-with-smallest-sums
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

import java.util.*;

class Solution {

    private int hash(int x, int y) {
        return (x + 1) * 100000 + y;
    }

    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        Set<Integer> visited = new HashSet<>();
        List<List<Integer>> ret = new ArrayList<>();
        PriorityQueue<int[]> queue = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return Integer.compare(nums1[o1[0]] + nums2[o1[1]], nums1[o2[0]] + nums2[o2[1]]);
            }
        });
        queue.offer(new int[]{0, 0});
        visited.add(hash(0, 0));
        while (k-- > 0 && !queue.isEmpty()) {
            int[] node = queue.poll();
            int x = node[0], y = node[1];
            ret.add(Arrays.asList(nums1[x], nums2[y]));

            if (x + 1 < nums1.length && !visited.contains(hash(x + 1, y))) {
                visited.add(hash(x + 1, y));
                queue.offer(new int[]{x + 1, y});
            }
            if (y + 1 < nums2.length && !visited.contains(hash(x, y + 1))) {
                visited.add(hash(x, y + 1));
                queue.offer(new int[]{x, y + 1});
            }
        }
        return ret;
    }
}