插入时间段 Insert Interval

题目: Given a set of  non-overlapping  intervals, insert a new interval into the intervals (merge if necessary).  You may assume that the intervals were initially sorted according to their start times.

Example 1: Given intervals [1,3],[6,9], insert and merge [2,5] in as [1,5],[6,9].

Example 2: Given [1,2],[3,5],[6,7],[8,10],[12,16], insert and merge [4,9] in as [1,2],[3,10],[12,16]. This is because the new interval [4,9] overlaps with [3,5],[6,7],[8,10].

思路:注意这里说了原时间段集合是不存在重叠的,因此,我只需要判断新添时间段对原时间段集合的影响。

1、只是新添加了一个时间段在其中,不影响原有的时间段;

2、影响了一个时间段;

2、合并了两个或多个时间段;

这样看来,只需要按顺序遍历一次有序的时间段集合即可。

/**
 * Definition for an interval.
 * struct Interval {
 *     int start;
 *     int end;
 *     Interval() : start(0), end(0) {}
 *     Interval(int s, int e) : start(s), end(e) {}
 * };
 */
class Solution {
public:
    vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {
        vector<Interval>::iterator it = intervals.begin();
        while(it != intervals.end())
        {
            if(newInterval.end < it->start)
            {
                intervals.insert(it, newInterval);
                return intervals;
            }
            else if(newInterval.start > it->end)
            {
                it++;
                
            }
            else //发现重叠
            {
                newInterval.start = min(newInterval.start, it->start);
                newInterval.end = max(newInterval.end, it->end);
                it = intervals.erase(it); //删除it所指结点,返回下一个结点位置
            }
        }
        intervals.insert(intervals.end(), newInterval);
        return intervals;
    }
};
上面的代码看似时间复杂度是O(N),实际上对vector的erase的每一次操作都是O(N)的,因此总的时间复杂度是O(N^2)。为了避免这样,不再原vector上做修改,而是新建一个新的vector作为新的时间段集合。以空间换取时间,这样时间和空间复杂度都为O(N)。

/**
 * Definition for an interval.
 * struct Interval {
 *     int start;
 *     int end;
 *     Interval() : start(0), end(0) {}
 *     Interval(int s, int e) : start(s), end(e) {}
 * };
 */
class Solution {
public:
    vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {
        vector<Interval> result;
        vector<Interval>::iterator it = intervals.begin();
        while(it != intervals.end())
        {
            if(newInterval.end < it->start)
            {
                break;
            }
            else if(newInterval.start > it->end)
            {
                result.push_back(*it);
                it++;
            }
            else //it结点和新时间段重叠,it结点被时间段影响
            {
                newInterval.start = min(newInterval.start, it->start);
                newInterval.end = max(newInterval.end, it->end);
                it++;
            }
        }
        result.push_back(newInterval);
        for(;it!=intervals.end();++it)
            result.push_back(*it);
        return result;
    }
};


附加:如果原时间段集合是杂乱无章的,还是用的以前的思路。把时间段拆成入和出两个,然后排序。得到有序的出入序列后按实时有无情况来生成新的时间段集合。为允许时间相同,使用了multimap。这个方法总感觉不太舒服。下一次想想更好的办法。

/**
 * Definition for an interval.
 * struct Interval {
 *     int start;
 *     int end;
 *     Interval() : start(0), end(0) {}
 *     Interval(int s, int e) : start(s), end(e) {}
 * };
 */
class Solution {
public:
    vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {
        
        multimap<int, int> newlist;
        newlist.insert(make_pair(newInterval.start, 1));
        newlist.insert(make_pair(newInterval.end, 0));
        vector<Interval>::iterator it = intervals.begin();
        for(;it != intervals.end();it++)
        {
            newlist.insert(make_pair(it->start, 1));
            newlist.insert(make_pair(it->end, 0));
        }

        
        intervals.clear();
        
        multimap<int, int>::iterator it2 = newlist.begin();
        int left, right, count;
        count = 0;
        left = it2->first;
        while(it2 != newlist.end())
        {
            if(it2->second == 1)
                count++;
            else
                count--;
            if(count == 0)
            {
                right = it2->first;
                it2++;
                if(it2 != newlist.end() && right == it2->first)
                {
                    it2++;
                    count++;
                    continue;
                }
                Interval tmp = Interval(left, right);
                intervals.push_back(tmp);
                if(it2 != newlist.end())
                    left = it2->first;
            }
            else
            {
                it2++;
            }
        }
        
        return intervals;
    }
};

你可能感兴趣的:(插入时间段 Insert Interval)