LeetCode[Array]: Insert Interval

来源:互联网 发布:网络招聘话术技巧 编辑:程序博客网 时间:2024/06/03 21:14

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].

看到这个题目,我首先产生的一个问题就是:是在原来的区间vector上修改,还是重新建一个然后返回?这个问题虽然不会导致算法根本上的差别,但是在实现上却会带来比较大的差别。

关于这个问题,我的想法是:由于传入的参数是引用类型,所以我认为不应该对原来的vector进行修改,而应该新建一个返回。

我的解法是设置两个标志foundStart和foundEnd分别标识是否找到新区间的起始点和终止点是否找到,纯粹地根据区间关系来确定,所以实现起来也相对比较麻烦:

vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {    vector<Interval> mergedIntervals;    int start = 0, end = 0;    bool foundStart = false, foundEnd = false;    for (auto interval : intervals) {        if (!foundStart) {            if (interval.end < newInterval.start) mergedIntervals.push_back(interval);            else if (interval.start > newInterval.end) {                mergedIntervals.push_back(newInterval);                mergedIntervals.push_back(interval);                foundStart = true;                foundEnd = true;            }            else {                start = interval.start < newInterval.start ? interval.start : newInterval.start;                foundStart = true;                if (newInterval.end <= interval.end) {                    mergedIntervals.push_back(Interval(start, interval.end));                    foundEnd = true;                }                else end = newInterval.end;            }        }        else if (!foundEnd) {            if (interval.start > end) {                mergedIntervals.push_back(Interval(start, end));                mergedIntervals.push_back(interval);                foundEnd = true;            }            else if (end < interval.end) {                mergedIntervals.push_back(Interval(start, interval.end));                foundEnd = true;            }        }        else mergedIntervals.push_back(interval);    }    if (!foundStart) mergedIntervals.push_back(newInterval);    else if (!foundEnd) mergedIntervals.push_back(Interval(start, end));    return mergedIntervals;}


在Discuss上有更好的解法。

0 0
原创粉丝点击