​LeetCode刷题实战632：最小区间

算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，这个公众号后续每天带大家做一道算法题，题目就从LeetCode上面选 ！

今天和大家聊的问题叫做 最小区间，我们先来看题面：

https://leetcode.cn/problems/smallest-range-covering-elements-from-k-lists/

You have k lists of sorted integers in non-decreasing order. Find the smallest range that includes at least one number from each of the k lists. We define the range [a, b] is smaller than range [c, d] if b - a < d - c or a < c if b - a == d - c.

你有 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] 中。

示例 2：

输入：nums = [[1,2,3],[1,2,3],[1,2,3]]
输出：[1,1]

解题

https://blog.csdn.net/Rutifrl/article/details/109343829

主要思路：

首先，最小区间要：

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

上期推文：

LeetCode1-620题汇总，希望对你有点帮助！

LeetCode刷题实战621：任务调度器

LeetCode刷题实战622：设计循环队列

LeetCode刷题实战623：在二叉树中增加一行

LeetCode刷题实战624：数组列表中的最大距离

LeetCode刷题实战625：最小因式分解

LeetCode刷题实战626：换座位

LeetCode刷题实战627：变更性别

LeetCode刷题实战628：三个数的最大乘积

LeetCode刷题实战629：K个逆序对数组

LeetCode刷题实战630：课程表 III

LeetCode刷题实战631：设计 Excel 求和公式