不会就坚持57天吧 和最小k树对

继续

转化为BFS
考虑nums1和nums2已经有序，显然nums1[0]于nums2[0]组成的数对和是最小的
后面简写为(0,0)，表示nums1取下标为0的元素，nums2取下标为0的元素组成数对
第2小的数对，应该从(1,0)和(0,1)中选择，也就是上一个最小数对(0,0),要么nums1的下标加1，要么nums2的下标加1
这类似于BFS：当访问到某个节点时，把它可达的节点入队列，作为后续的选择
具体地，当(u,v)为第x小的数对时，第x+1小的数对可能是(u+1,v)和(u,v+1)，但不限于这两者
使用优先队列得到最小数对
上述bfs方法计算出的候选最小数对加入队列，队列中存在的数对都有可能是第x+1小的数对。
若直接遍历队列中的所有数对选出最小，固然可行，但时间复杂度有点高，为O(队列长度)
很容易想到，若bfs加入的队列是一个优先队列（最小堆），那么队首直接就是最小数对了
加入优先队列的时间复杂度为O(log队列长度），弹出队首元素的时间复杂度也是O(log队列长度)
为什么有重复&去重方法
重新观察具体的最小数对，前面分析过，第1小的是(0.0)，第2小的可能是(0,1),(1,0)
假设第2小的是(0,1)，那么(0,2),(1,1)会加入优先对列中，此时优先队列中有(1,0),(0,2),(1,1)，暂时没问题
假设第3小的是(1,0)，那么(2,0),(1,1)会加入优先对列中，此时优先队列中有(0,2),(1,1),(2,0),(1,1)，出现重复数对
可以使用HashSet去重，将字符串"u_v"作为key，访问过的key加入set中，若set中已存在key，则不加入优先队列
上面的例子，第2步会在set中加入"1_1"，此时(1,1)能进入优先队列
第3步也会尝试在set中加入"1_1"，但加入失败，此时(1,1)不进入优先队列

class Solution {
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        PriorityQueue<int[]> pq = new PriorityQueue<int[]>((a,b)->nums1[a[0]]+nums2[a[1]]-nums1[b[0]]-nums2[b[1]]);
        HashSet<String> set = new HashSet<String>();
        pq.offer(new int[]{0,0});
        List<List<Integer>> ans = new ArrayList<>();
        while(k-->0&&pq.size()>0)
        {
            int[] pair = pq.poll();
            ans.add(Arrays.asList(nums1[pair[0]],nums2[pair[1]]));
            if(pair[0]+1<nums1.length)
            {
                String key = String.valueOf(pair[0]+1)+"_"+String.valueOf(pair[1]);
                if(set.add(key))
                {
                    pq.offer(new int[]{pair[0]+1,pair[1]});
                }
            }
            if(pair[1]+1<nums2.length)
            {
                String key = String.valueOf(pair[0])+"_"+String.valueOf(pair[1]+1);
                if(set.add(key))
                {
                    pq.offer(new int[]{pair[0],pair[1]+1});
                }
            }
        }
        return ans;
    }
}