继续
转化为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;
}
}