你有 k 个升序排列的整数数组。找到一个最小区间,使得 k 个列表中的每个列表至少有一个数包含在其中。
我们定义如果 b-a < d-c 或者在 b-a == d-c 时 a < c,则区间 [a,b] 比 [c,d] 小。
示例:
输入:[[4,10,15,24,26], [0,9,12,20], [5,18,22,30]]
输出: [20,24]
解释:
列表 1:[4, 10, 15, 24, 26],24 在区间 [20,24] 中。
列表 2:[0, 9, 12, 20],20 在区间 [20,24] 中。
列表 3:[5, 18, 22, 30],22 在区间 [20,24] 中。
题解:这道题的关键在于题目的理解上,将题目换个说法,就会明朗很多 :每个数组中找一个数,让最小数与最大数的差值最小
import java.util.*;
class Solution {
public int[] smallestRange(List<List<Integer>> nums) {
//保存每个数组的指针位置
int[]index=new int[nums.size()];
/*
*构造一个小顶堆,每个插入的数都会自动按照升序排列
*/
PriorityQueue<Integer>Range=new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return nums.get(o1).get(index[o1])-nums.get(o2).get(index[o2]);
}
});
/*
*max记录堆中的最大数
*首先每个数组都往堆中填充一个数字
*/
int max=Integer.MIN_VALUE;
for(int i=0;i<nums.size();i++){
Range.add(i);
max=Math.max(max,nums.get(i).get(index[i]));
}
/*
*minIndex记录堆中最小的数字的索引
*minRange记录当前最小的范围长度
*resLeft与resRight记录左右边界
*/
int minIndex=-1;
int minRange=Integer.MAX_VALUE;
int resLeft=0;
int resRight=0;
/*
*始终保证堆中每个数组都有一个元素在其中
*每次弹出最小的元素,同时插入最小元素所属数组的一个元素
*当某一个数组的指针到达终点后,不能再维持堆中每个数组都有一个元素在其中
*/
while(true){
minIndex= Range.poll();
if(max-nums.get(minIndex).get(index[minIndex])<minRange){
resLeft=nums.get(minIndex).get(index[minIndex]);
resRight=max;
minRange=resRight-resLeft;
}
index[minIndex]++;
if(index[minIndex]==nums.get(minIndex).size())
break;
Range.add(minIndex);
max = Math.max(nums.get(minIndex).get(index[minIndex]), max);
}
return new int[]{resLeft,resRight};
}
}