力扣57-插入区间-C++

一、题目

给你一个 无重叠的 ，按照区间起始端点排序的区间列表。

在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。

示例 1：

输入：intervals = [[1,3],[6,9]], newInterval = [2,5]
输出：[[1,5],[6,9]]
示例 2：

输入：intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
输出：[[1,2],[3,10],[12,16]]
解释：这是因为新的区间 [4,8] 与 [3,5],[6,7],[8,10] 重叠。
示例 3：

输入：intervals = [], newInterval = [5,7]
输出：[[5,7]]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/insert-interval

二、思路

输入区间数组 intervals 已经是排好序的，可以之间从头到尾遍历intervals，如果当前区间的左边界比newinterval 左边界小，则之间将该区间插入到结果数组ans中，知道当期区间的左边界比newinterval 的左边界大或interval 的全部区间已经插入ans 中，则将 newinterval 插入ans中，插入newinterval 时会出现两种情况：

1）若 newinterval 的左边界比ans 当前右边界大，则无重叠，直接插入ans中；

2）若有重叠，则修改ans 当前最后一个区间的右边界为两者中的最大值。

intervals中剩余的区间做同样处理，若无重叠，则直接插入，若有重叠，则修改ans 当前最后一个区间的右边界为两者中的较大值。

三、代码

class Solution {
public:
    vector<vector<int>> insert(vector<vector<int>>& intervals, vector<int>& newInterval) {
        vector<vector<int>> ans;
        int len = intervals.size();
        if(len == 0 ){
            ans.push_back(newInterval);
            return ans;
        }
        int i=0, index = 0; //i为intervals下标，index为ans下标
        while(i < len && intervals[i][0] <= newInterval[0]){ 
            ans.push_back(intervals[i]);
            i++; index++;
        }
        if(i == len){   //intervals的全部区间已经插入ans中
            if(newInterval[0] > ans[index-1][1]){
                ans.push_back(newInterval);
                index++;
            }else{
                ans[index-1][1] =newInterval[1] > ans[index-1][1] ? newInterval[1] : ans[index-1][1];
            }
            return ans;
        }
        if(index > 0 && newInterval[0] > ans[index-1][1] || index == 0){  //插new入区间
            ans.push_back(newInterval);
            index++;
        }else{   //区间重叠
            ans[index-1][1] = newInterval[1] > ans[index-1][1] ? newInterval[1] : ans[index-1][1];
        }
        
        while(i < len){
            if(intervals[i][0] > ans[index-1][1]){
                ans.push_back(intervals[i]);
                index++;  i++;
            }else{   //区间重叠
                ans[index-1][1] = intervals[i][1] > ans[index-1][1] ? intervals[i][1] : ans[index-1][1];
                i++;
            }
        }
        return ans;
    }
};

四、运行结果