你有 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>。重置代码模板后可以看到这项改动。
这个题的官方题解有些晦涩难懂,这里打算在官方题解的基础之上加一些个人的理解。
根据题意,这个题目可以转化为从k个列表中各取一个数,使得这k个数中的最大值与最小值的差最小。(也就是,我们只需要找到这个最大值和最小值就可以了)
假设这k个数中的最小值是第i个列表中的x,对于任意的j != i,设第j个列表中被选中的数是y,那么为了找到最小区间,y应该取第j列表中大于等于x的最小的数。至于证明,我们可以用Cut and Paste的方法来证明。所以,其余k - 1个列表中应该取大于等于x的最小的数。(我们根据这个性质来设置右边界)
为了优化,这里用了小顶堆和指针,而且指针是用数组的方式实现的。
具体实现逻辑可结合代码注释体会一下。
class Solution {
public:
vector smallestRange(vector>& nums) {
int rangeLeft = 0, rangeRight = INT_MAX;
int size = nums.size();
//next数组起指针的作用
vector next(size);
auto cmp = [&](const int &u, const int &v){
return nums[u][next[u]] > nums[v][next[v]];
};
//维护一个小顶堆
priority_queue, decltype(cmp)> pq(cmp);
int minValue = 0, maxValue = INT_MIN;
for(int i = 0; i < size; i++){
pq.emplace(i);
//一开始的右边界是所有最小值中的最大值
maxValue = max(maxValue, nums[i][0]);
}
while(true){
//枚举左边界
int row = pq.top();
pq.pop();
minValue = nums[row][next[row]];
//更新最小区间的长度
if(maxValue - minValue < rangeRight - rangeLeft){
rangeLeft = minValue;
rangeRight = maxValue;
}
//说明有一个列表已经枚举完,不能再选中其中的数,所以退出
if(next[row] == nums[row].size() - 1) break;
//此时next[row]已经被废弃,该列表的下一个数是next[row] + 1
++next[row];
//时刻更新右边界,保证每个列表中都有数被选中
//而且,按照这种方式更新,可保证右边界是满足条件下的最小的
maxValue = max(maxValue, nums[row][next[row]]);
pq.emplace(row);
}
return {rangeLeft, rangeRight};
}
};