题目链接: https://leetcode.com/problems/insert-interval/description/
Description
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]
.
解题思路
题意大致为在一个已经排序过的相互无交集的区间数组中,添加一个新的区间,将有交集的区间全部合并成一个新的区间。
题目不是很难,注意下细节就可以了。遍历初始区间数组,对每一个区间判断与目标区间是否有交集,交集的判定可以归结为三种,两种括号表示两个不同的区间
(---<--)--->
<---(-->---)
<---(--)--->
有交集的区间个数可以通过另一个迭代器来记录,每发现一个区间有交集迭代器就加一,直到无交集或者到达数组末尾为止,删除该数组区间,再在此位置插入新的区间。
另外,还要考虑新区间与原来数组中的区间都没有交集的情况。通过比较新区间的 end
是否小于该轮第一个区间的 start
来判断是否有交集,涵盖了每个原数组区间前面的空隙,不包括最后一个区间后面的间隙,因此还要单独比较新区间 start
和原数组中最后一个区间的 end
。
Code
class Solution {
public:
vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {
if (intervals.empty()) {
intervals.push_back(newInterval);
} else {
for (auto it = intervals.begin(); it != intervals.end(); ++it) {
auto eit = it;
int start = newInterval.start;
int end = newInterval.end;
while (eit != intervals.end()) {
if ((start <= eit->end && start >= eit->start) ||
(end >= eit->start && end <= eit->end) ||
(start <= eit->start && end >= eit->end)) {
start = min(start, eit->start);
end = max(end, eit->end);
eit++;
} else {
break;
}
};
if (eit != it || end < it->start) {
it = intervals.erase(it, eit);
intervals.insert(it, Interval(start, end));
break;
} else if (it + 1 == intervals.end() && start > it->end) {
intervals.insert(it + 1, Interval(start, end));
break;
}
}
}
return intervals;
}
};