leetcode 57. 插入区间

题目:

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

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

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

解析:

这里可以先将目标区间插入正确的位置, 然后再单独进行合并操作, 时间复杂度O(n), 空间复杂度O(n)

class Solution {
public:
    vector<Interval> insert(vector<Interval>& intervals, Interval newInterval) {
        int i = 0;
        for(; i < intervals.size();++i) 
        {
            if(intervals[i].start >= newInterval.start )
            {
                intervals.insert(intervals.begin() + i, newInterval);
                break;
            }
        }
        if(i == intervals.size())
            intervals.push_back(newInterval);
        return move(merge(intervals));
    }
    vector<Interval> merge(vector<Interval>& intervals)
    {
        vector<Interval> res;
        for(auto i : intervals)
        {
            if(res.size() < 1 || res.back().end < i.start)
                res.push_back(i);
            else
            {
                res.rbegin()->end = max(res.rbegin()->end, i.end);
            }
        }
        return move(res);
    }
};