632. 最小区间

最新推荐文章于 2022-11-20 15:35:13 发布

xxZkj

最新推荐文章于 2022-11-20 15:35:13 发布

阅读量168

点赞数

分类专栏： Leecode

本文链接：https://blog.csdn.net/weixin_38688399/article/details/107729316

版权

Leecode 专栏收录该内容

152 篇文章 0 订阅

订阅专栏

该博客讨论了如何解决632号问题——找到覆盖k个升序数组的最小区间。通过堆数据结构，博主提出了一种策略，即维护一个最小堆来跟踪每个列表的最小值，并同时记录堆中元素的最大值。每次从堆中取出最小值，计算当前区间，与已知最小区间比较并更新。然后将对应列表的指针右移，将新元素加入堆并更新堆中元素的最大值。

摘要由CSDN通过智能技术生成

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.second;
    }
};

class Solution {
public:
    vector<int> smallestRange(vector<vector<int>>& nums) {
        int leftRange = 0, rightRange = INT_MAX;
        int size = nums.size();
        priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pq;
        //维护k个列表的访问位置
        vector<int> next(size, 0);
        
        int minValue = 0, maxValue = INT_MIN;
        //初始状态为k个列表的0号元素
        for(int i = 0; i < size; i++) {
            pq.push({i, nums[i][0]});
            maxValue = max(maxValue, nums[i][0]);
        }

        while(true) {
            pair<int, int> tmp = pq.top();
            pq.pop();
            //取出最小值
            minValue = tmp.second;
            //更新区间
            if(maxValue - minValue < rightRange - leftRange) {
                leftRange = minValue;
                rightRange = maxValue;
            }
            //一个列表遍历完毕
            if(next[tmp.first] == nums[tmp.first].size() - 1){
                break;
            }
            //最小值列表所在元素位置前进
            ++next[tmp.first];
            //新加入元素，最大值需重新维护
            maxValue = max(maxValue, nums[tmp.first][next[tmp.first]]);
            //将新元素加入堆
            pq.push({tmp.first, nums[tmp.first][next[tmp.first]]});
        }
        return {leftRange, rightRange};
    }
};