LeetCode: Insert Interval

思路:其实和前面一题完全一样,不过可以先要讲新区建插入到已有的区间数组中,保持节点首节点有序,插入完以后,后面就和前面一题一样了。

考虑一下不插入区间节点,发现需要考虑好多复杂的问题,具体看代码注释吧,写了好久,最后代码已经复杂得惨不忍睹了,难道真的这么难吗?以后再来考虑这题,先贴代码。

code:

class Solution {
public:
    vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {
        vector<Interval> ret;
        if(intervals.empty())    {
                ret.push_back(newInterval);
                return ret;
        }
        int flag = 0;  // 标记是否合并好了区间,flag = 0 表示没有开始合并,flag = 1表示准备合并了,已经确定好合并后的首节点了,flag = 2 表示合并好了区间且存入结果数组,flag = 3表示不经过合并存入新区间(即新区间和已有区间集没有交集)。
        Interval mergeInterval(newInterval.start,newInterval.end);
        if(mergeInterval.end < intervals[0].start){
            ret.push_back(mergeInterval);
            flag = 2;
        }
        for(int i=0;i<intervals.size();i++){
            if(i > 0 && mergeInterval.end < intervals[i].start && mergeInterval.start > intervals[i-1].end){
                ret.push_back(mergeInterval);
                flag = 3;
            }
            if(newInterval.start <= intervals[i].end && newInterval.end >= intervals[i].start){
                mergeInterval.start = min(mergeInterval.start,intervals[i].start);
                flag = 1;
            }
            if(flag == 1 && newInterval.end <= intervals[i].end){
                if(newInterval.end >= intervals[i].start){
                    mergeInterval.end = intervals[i].end;
                    ret.push_back(mergeInterval);
                }
                else
                    ret.push_back(mergeInterval);
                flag = 2;
            }
            if(flag == 0 || flag == 3 || (flag == 2 && newInterval.end < intervals[i].start))
                ret.push_back(intervals[i]);
        }
        if(flag != 2 && flag != 3) 
            ret.push_back(mergeInterval);
        return ret;
        
    }
};



你可能感兴趣的:(LeetCode: Insert Interval)