题目描述:
你有 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] 中。
注意:
给定的列表可能包含重复元素,所以在这里升序表示 >= 。
1 <= k <= 3500
-105 <= 元素的值 <= 105
对于使用Java的用户,请注意传入类型已修改为List<List>。重置代码模板后可以看到这项改动。
方法1:哈希函数配合双指针
主要思路:
(1)将原数组转为哈希表,表的键使用数组中的值,对应的值使用vector数组存储该键出现在的各个数组时,数组的索引,同时统计这些出现的键的最大值和最小值;
(2)在这些键中使用双指针的滑动窗口,来找能够覆盖所有的数组的索引的最小的窗口;
class Solution {
public:
vector<int> smallestRange(vector<vector<int>>& nums) {
//将原数组转化为哈希表
unordered_map<int,vector<int>> mp;
//统计所有数组中的元素的最大值和最小值,以确定初始窗口的大小
int min_num=INT_MAX;
int max_num=INT_MIN;
//统计转化
for(int i=0;i<nums.size();++i){
for(int j=0;j<nums[i].size();++j){
mp[nums[i][j]].push_back(i);
min_num=min(min_num,nums[i][j]);
max_num=max(max_num,nums[i][j]);
}
}
//设置左右窗口的初始值和滑动窗口的初始值
int right=min_num-1;//这个先减去1,是为了统一循环中的++right操作
int left=min_num;
int win_left=min_num;
int win_right=max_num;
int count=0;//统计出现过的数组的索引是否满足要求
vector<int> freq(nums.size());//统计各个数组索引出现的频率
while(right<max_num){
++right;//扩大右侧窗口
if(mp.count(right)){//若该右侧窗口值出现过,则统计其对应的各个数组的索引的出现频率
for(int& i:mp[right]){
freq[i]+=1;
if(freq[i]==1){//若频率等于 1 时,统计一次
++count;
}
}
}
while(count==nums.size()){//此时的窗口是否已经覆盖了所有的数组的索引
if(right-left<win_right-win_left){//判断是否更新最小窗口的大小
win_left=left;
win_right=right;
}
if(mp.count(left)){//判断是否更新左侧窗口对应的各个数组的索引
for(int& i:mp[left]){
freq[i]-=1;
if(freq[i]==0){//若某个数组的索引出现的次数降为0,则将窗口覆盖的大小减小1
--count;
}
}
}
++left;//更新左侧窗口的值
}
}
return {win_left,win_right};//返回最小的窗口
}
};