力扣-057 插入区间

题目描述

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

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

样例

示例 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] 重叠。

分析

这道题跟上道题类似，当然，你也可以直接在后面增加一个区间，然后按照上一题的来做，但这里题目已经给出，给出的是一个无重叠的 ，按照区间起始端点排序的区间列表，所以我们可以直接在里面合并。
首先，我们需要找出第一个与插入区间相交的区间，即第一个$intervals[i][1]>newIntervals[0]$，然后进入下一个状态，继续往下找，找出第一个与插入区间不相交的区间，即$intervals[j][0]>newIntervals[1]$，那么合并的区间变成$[min(intervals[i][0],newIntervals[0]),max(intervals[j-1][1],newIntervals[1]))$，其余不相交的区间直接按顺序插入就行，这里需要处理一下边界情况。也是建议画图，这样理解起来比较直观。
时间复杂度为$O(n)$

代码

class Solution {
public:
    vector<vector<int>> insert(vector<vector<int>>& intervals, vector<int>& newInterval) {
        vector<vector<int>> ans;
        int len = intervals.size();
        int flag = 0;
        int p, q = newInterval[1];
        for (int i = 0; i < len; i++) {
            if (newInterval[0] > intervals[i][1]) {
                ans.push_back(intervals[i]);
            } else {
                if (flag == 0) {
                    p = min(newInterval[0], intervals[i][0]);
                    flag = 1;
                }
            }
            if (newInterval[1] < intervals[i][0]) {
                if (flag == 1) {
                    ans.push_back({p, q});
                }
                ans.push_back(intervals[i]);
                flag = 2;
            } else {
                q = max(q, intervals[i][1]);
            }
        }
        if (flag == 1) {
            q = max(q, intervals.back()[1]);
            ans.push_back({p, q});
        }
        if (flag == 0) {
            ans.push_back(newInterval);
        }
        return ans;
    }
};