你有 k 个 非递减排列 的整数列表。找到一个 最小 区间,使得 k 个列表中的每个列表至少有一个数包含在其中。
我们定义如果 b-a < d-c 或者在 b-a == d-c 时 a < c,则区间 [a,b] 比 [c,d] 小。
示例 1:
输入:nums = [[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.最短
2.在长度相等时,起点最小
为了满足2,取所有列表的最小值放入区间中,又要使区间变短,那这个区间的最小值应该往右移,也就是把列表中,这个值的下一位拿出来替代它,在更新的过程中,记录区间的长度,就能找到最小区间。
用一个最小堆存放元素(num,a,b),a为num所在的列表下标,b为num在列表中的下标,最小值为堆顶元素,用一个变量记录最大值
class Solution {
private:
struct node
{
int val, indexa, indexb;
//记录当前元素的值和它在哪个区间,区间的哪个位置
node(int v, int a, int b):val(v), indexa(a), indexb(b){}
};
friend bool operator > (const struct node& a, const struct node& b)
{
return (a.val > b.val);
};
public:
vector<int> smallestRange(vector<vector<int>>& nums) {
//区间的最大值
int mx = nums[0][0];
//最小区间的长度
int dis = 200000;
int mxindexa = 0, mxindexb = 0;
//最小区间的左右端点
int left, right;
//auto cmp = [](node left, node right){return left.val > right.val;};
//建立最小堆
priority_queue<node, vector<node>, greater<node> > pq;
//每个元素的最小值入堆,构成初始区间
for(int i = 0; i < nums.size(); i++)
{
pq.push(node(nums[i][0], i, 0));
//记录最大值和它所在的区间
if(mx < nums[i][0])
{
mx = nums[i][0];
mxindexa = i;
}
}
node cur = pq.top();
//替代区间最小值的下一个元素
int next;
while(1)
{
//更换掉当前区间的最小值
cur = pq.top();
pq.pop();
//更新最小区间
if(mx - cur.val < dis)
{
dis = mx - cur.val;
left = cur.val;
right = mx;
}
//如果当前元素已经是所在列表的末尾
if(cur.indexb + 1 >= nums[cur.indexa].size()) break;
//用列表中元素的下一个元素替代它
next = nums[cur.indexa][cur.indexb + 1];
pq.push(node(next, cur.indexa, cur.indexb + 1));
if(mx < next) mx = next;
}
return {left, right};
}
};