原题链接:
力扣https://leetcode.cn/problems/insert-interval/
这题比56合并区间稍微复杂一些。给定新区间,让插入到一个有序二维数组。注意给定的二维数组是无重叠的,所以我们只需要遍历给定二维数组,把新区间左右端点与二维数组中每个数组左右端点对比即可。
关键是分清三种情况,设置标志placed记录新区放没放过。
1.新区在当前区间左边(如果没放过新区就放,然后插入当前区间)
2.新区在当前区间右边(直接插入当前区间)
3.新区与当前区间有交集(计算新的新区左端点右端点)。
最后如果新区没放过,说明在所有给定区间右侧,直接放入。
具体代码与注释:
class Solution {
public:
vector<vector<int>> insert(vector<vector<int>>& intervals, vector<int>& newInterval) {
int left = newInterval[0];//要插入的数组的左右值
int right = newInterval[1];
bool placed = false;
vector<vector<int>> ans;//创建一个新二维容器存放答案
for (const auto& interval: intervals) {//for(const auto &a:b),a不可修改,用于只读取b中内容
if (interval[0] > right) {// 在插入区间的右侧且无交集
if (!placed) {//如果新区没放过!
ans.push_back({left, right});
placed = true;//标志改变!
}
ans.push_back(interval);
}
else if (interval[1] < left) {
// 在插入区间的左侧且无交集
ans.push_back(interval);
}
else {// 与插入区间有交集,计算它们的并集
left = min(left, interval[0]);//注意哪个取小哪个取大!
right = max(right, interval[1]);
}
}
if (!placed) {//如果新区间没放过!
ans.push_back({left, right});
}
return ans;
}
};