刷题升级之路：Leetcode——757.设置交集大小至少为2

题目描述

一个整数区间 [a, b] ( a < b ) 代表着从 a 到 b 的所有连续整数，包括 a 和 b。

给你一组整数区间intervals，请找到一个最小的集合 S，使得 S 里的元素与区间intervals中的每一个整数区间都至少有2个元素相交。
输出这个最小集合S的大小。

示例 1:
输入: intervals = [[1, 3], [1, 4], [2, 5], [3, 5]]
输出: 3
解释:
考虑集合 S = {2, 3, 4}. S与intervals中的四个区间都有至少2个相交的元素。
且这是S最小的情况，故我们输出3。

示例 2:
输入: intervals = [[1, 2], [2, 3], [2, 4], [4, 5]]
输出: 5
解释:
最小的集合S = {1, 2, 3, 4, 5}.

注意:
intervals 的长度范围为[1, 3000]。
intervals[i] 长度为 2，分别代表左、右边界。
intervals[i][j] 的值是 [0, 10^8]范围内的整数。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/set-intersection-size-at-least-two

算法思路与分析

见到此题第一反应是用贪心策略解决（此题是一个最优化问题，在区间上选取一些整数点以满足最优条件，理想的情况是贪心策略管用，做的题多一些之后看到这种形式的题目自然会想到贪心）。

那么接下来我们思考一下，用贪心法是否能解决呢？

首先将无序的输入区间整理为有序序列，根据区间的右端点从小到大排序，在此基础上进行贪心法的操作。
因为要选最少的数，目标是满足至少2个数的重叠要求，直觉上当然是不得不取时才取数，不去是多余的数，这样总体上取得尽量少。

我们已经排好序，就从第一个区间[a1,b1]（即右端点最小的区间）开始此过程。在这个区间上必须有两个数取到，最好的选择一定是最靠右的两数：b1-1和b1。可以证明在第一步时两个数是最优选择，且越靠右越有可能用于重叠更多的集合，在后续过程里一定不比其他选择更差。

我的想法是维护前k个区间[a1,b1],…,[ak,bk]已选取的最大的两个数last1和last2(last1>last2)，考虑新的区间[a(k+1),b(k+1)]，如果这两个数并非都在新的区间内，需要选新的数：
（1）如果新区间的左端点a(k+1)>last1（出现“断层”），在新区间选择两个数，和[a1,b1]的情况一样，选b(k+1)-1和b(k+1)
（2）如果新区间的左端点last2<a(k+1)<=last1，在新区间选择一个数，需要分两种情况考虑（这部分比较容易考虑不全，我也是提交错误之后看了示例才想到…）：
1.last1=b(k+1)，即上个区间选了右端点且与新区间右端点重合，此时选择b(k+1)-1
2.else，选择最靠右的数b(k+1)
在过程中一定要不断更新last1和last2哦

以上算法可以通过对步数k归纳证明是最优解，当前k步依据此算法得到最优解，第k+1步的解依然是最优的。具体过程不在此写明，读者可以自行练习。

代码实现

class Solution {
public:
    struct op{
        bool operator()(const vector<int>& a, const vector<int>& b){
            if(a[1]==b[1])
                return a[0]<b[0];
            return a[1]<b[1];
        }
    };
    int intersectionSizeTwo(vector<vector<int>>& intervals) {
        int n = intervals.size();
        sort(intervals.begin(),intervals.end(),op());
        // for(int i=0;i<n;i++){
        //     cout<<'['<<intervals[i][0]<<','<<intervals[i][1]<<']'<<'\t';
        // }
        // cout<<endl;
        int ans = 2;
        int last1 = intervals[0][1], last2 = intervals[0][1]-1;   //当前最大的两个选取数
        //  cout<<last1<<"\t"<<last2<<endl;
        for(int i = 1; i < n; i++){
            if(intervals[i][0] > last1){
                last1 = intervals[i][1];
                last2 = intervals[i][1]-1;
                ans += 2;
            }
            else if(intervals[i][0] <= last1){
                if(intervals[i][0] > last2){
                    if(intervals[i][1]!=last1){
                        last2 = last1;
                        last1 = intervals[i][1];
                        
                    }
                    else{
                        last2 = last1-1;

                    }
                    ans++;
                }
            }
            // cout<<last1<<"\t"<<last2<<endl;
        }
        return ans;
    }
};