你有 k 個升序排列的整數數組。找到一個最小區間,使得 k 個列表中的每個列表至少有一個數包含在其中。
我們定義如果 b-a < d-c 或者在 b-a == d-c 時 a < c,則區間 [a,b] 比 [c,d] 小。
示例 1:
輸入:[[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] 中。
代碼
class Solution {class numG{int g,num;public numG(int g,int num){this.g=g;this.num=num;}}public int[] smallestRange(List<List<Integer>> nums) {PriorityQueue<numG> priorityQueue=new PriorityQueue<>(((o1, o2) -> o1.num-o2.num));
//維護最小堆int end=Integer.MIN_VALUE;int[] index=new int[nums.size()];//記錄每個序列的指針for(int i=0;i<nums.size();i++){if(nums.get(i).get(0)>end) end=nums.get(i).get(0);//找出起始區間numG temp=new numG(i,nums.get(i).get(0));priorityQueue.offer(temp);//將每個序列首元素入隊}int max=end;int start=priorityQueue.peek().num;int min=start;//初始區間int len=end-start+1;//區間長度while (true){int gr=priorityQueue.poll().g;//將最小的數值出隊if (index[gr]+1==nums.get(gr).size()) break;//已經到了該序列的末尾了index[gr]++;//指向該序列的下一個numG n=new numG(gr,nums.get(gr).get(index[gr]));//將下一個入隊priorityQueue.offer(n);max= Math.max(max,n.num);//區間擴大min=priorityQueue.peek().num;if(max-min+1<len)//如果是更短的區間,則替換掉原來的區間{start=max;end=min;len=max-min+1;}}return new int[]{end,start};}
}