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

思路:在56题的基础上,先插入区间,再合并区间,思路非常简单。

class Solution {
public:
    static bool cmp(Interval a, Interval b){
        if (a.start != b.start)
            return a.start < b.start;
        else
            return a.end < b.end;
    }

    //合并区间
    vector<Interval> merge(vector<Interval>& intervals) {
        vector<Interval> re;
        re.push_back(intervals[0]);
        for (int i = 1; i < intervals.size(); ++i){
            if (intervals[i].start <= re.back().end)
                re.back().end = max(re.back().end, intervals[i].end);
            else
                re.push_back(intervals[i]);
        }
        return re;
    }

    //先插入区间
    vector<Interval> insert(vector<Interval>& intervals, Interval newInterval) {
        sort(intervals.begin(), intervals.end(), cmp);
        if (intervals.empty() || intervals.back().start < newInterval.start)
            intervals.push_back(newInterval);
        else{
            for (int i = 0; i < intervals.size(); ++i){
                if (intervals[i].start >= newInterval.start){
                    intervals.insert(intervals.begin() + i, newInterval);
                    break;
                }
            }
        }
        return merge(intervals);
    }
};

猜你喜欢

转载自blog.csdn.net/scarlett_guan/article/details/80250156