632. 最小区间
你有 k 个升序排列的整数数组。找到一个最小区间,使得 k 个列表中的每个列表至少有一个数包含在其中。
我们定义如果 b-a < d-c 或者在 b-a == d-c 时 a < c,则区间 [a,b] 比 [c,d] 小。
思路
堆
问题转化为从k个列表中各取一个数字,保证这k个数的最大值和最小值之差最小。
使用一个最小堆维护当前k个指针指向元素的最小值,同时维护堆元素的最大值,每次从堆中取出最小值,根据最大值和最小值计算当前区间,如果当前区间小于最小区间则用当前区间更新最小区间,然后将对应列表的指针右移,将新元素加入堆中,并更新堆中元素的最大值。
代码
//重载仿函数,维护最小堆
struct cmp {
bool operator() (const pair<int, int>& a1, const pair<int, int>& a2) {
if(a1.second == a2.second) {
return a1.first > a2.first;
}
return a1.second > a2