Leetcode刷题笔记 632. 最小区间

题目
你有 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
-10⁵ <= 元素的值 <= 10⁵
解法

1. 我们把nums中的元素放入一个数组中，并标记元素来自哪个列表，可以得到

 nums中所有元素排序：0 4 5 9 10 12 15 18 20 22 24 26 30
    元素来自哪个列表：1 0 2 1 0  1  0  2  1  2  0  0  2

2. 要求找到一个最小区间，使得每个列表中至少有一个数包含在其中，也就是找到一个区间[a,b]，a、b两个元素之间要包含 所有 不同列表的值(即0、1、2)
3. 当 a=0,b=5时，这个区间包含来自第1、0、2列列表的值，是其中一个区间（不是最小的）；而当a=10,b=15时，其中只包含来自0、1列表值，不是一个区间；当a=15,b=22时,也包含第0、2、1、2个列表中的值。
4. 可以用滑动窗口来解决，这些窗口中要包含所有列（0，1，2）
5. 用 left 和 right 两个指针分别表示其左右边界，从左到右移动 left 和 right 指针，就像是一个窗口在序列上进行滑动，不断的有新元素从右侧进入窗口，同时有旧的元素从左侧移除窗口，当前窗口内的元素是否满足需求制约着窗口的扩张与收缩。

代码

#include <stdio.h>
#include <vector>
#include <iostream>
#include <unordered_map>
using namespace std;
class Solution {
public:
    vector<int> smallestRange(vector<vector<int>>& nums) {
        //放入数组中，并排序
        int k = nums.size();//k个列表
        vector<pair<int, int>> map;
        for (int i=0;i<k;i++){
            for(int j=0;j<nums[i].size();j++){
                map.push_back({nums[i][j],i});
            }
        }
        sort(map.begin(),map.end());
        
        vector<int> flag(k);//标记区间中是否有k个列表中的元素
        vector<int> ans;
        int left = 0,n= map.size(),cnt=0;
        int dif = INT_MAX;
        //往右滑动窗口
        for (int right = 0; right < n; right ++) {
            //判断这个元素来自哪个列表，如第一次出现，标记在[left,right]区间中，这个列表中出现过元素了。
            if (flag[map[right].second] == 0) cnt ++;
            flag[map[right].second] ++;
            //k个不同的列表中，每个列表中至少有一个元素在区间出现过了，寻找最小区间
            while (cnt == k && left <= right) {
                if (dif > map[right].first - map[left].first) {
                    dif = map[right].first - map[left].first;
                    ans = {map[left].first, map[right].first};
                }
                //向左移动
                flag[map[left].second]--;
                if(flag[map[left].second] == 0)
                    cnt--;
                left++;
            }
        }
        return ans;
    }
};
int main()
{
    vector<vector<int>> d(3, vector<int> ());
    d[0] = {4,10,15,24,26};
    d[1] = {0,9,12,20};
    d[2] = {5,18,22,30};
    Solution s;
    vector<int> ans = s.smallestRange(d);
    cout<<ans[0]<<" "<<ans[1];
    return 0;
}

今天也是爱zz的一天哦！