本文已参与「新人创作礼」活动，一起开启掘金创作之路。
插入区间
57. 插入区间 - 力扣（LeetCode） (leetcode-cn.com)
给你一个 无重叠的 ，按照区间起始端点排序的区间列表。
在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。
示例 1：
输入：intervals = [[1,3],[6,9]], newInterval = [2,5]
输出：[[1,5],[6,9]]
示例 2：
输入：intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
输出：[[1,2],[3,10],[12,16]]
解释：这是因为新的区间 [4,8] 与 [3,5],[6,7],[8,10] 重叠。
示例 3：
输入：intervals = [], newInterval = [5,7]
输出：[[5,7]]
示例 4：
输入：intervals = [[1,5]], newInterval = [2,3]
输出：[[1,5]]
示例 5：
输入：intervals = [[1,5]], newInterval = [2,7]
输出：[[1,7]]
提示：
- 0 <= intervals.length <= 10^4
- intervals[i].length == 2
- 0 <= intervals[i][0] <= intervals[i][1] <= 10^5
- intervals 根据 intervals[i][0] 按 升序 排列
- newInterval.length == 2
- 0 <= newInterval[0] <= newInterval[1] <= 10^5
解题思路
结果必定有一个新区间产生，有两种情况：newInterval本身，或者newInterval和若干个原区间合并成的区间。
给新区间设定一下属性index【插入的位置】，start、end【新区间值】，count【被合并的区间数量】。
- 初始化属性值，此时默认原数组长度为0，即intervals.length == 0；
- 遍历intervals，更新属性值。因为初始化时默认原数组长度为0，此时只需考虑长度大于0的情况；
- intervals[0][0] > newInterval[1]时前面插入，无交集
- intervals[intervals.length-1][1] < newInterval[0]时后面插入，无交集
- 其它为中间插入，flag被用来标记合并是否开始，具体看代码
- 根据原数组和新区间属性值，组装结果；count的存在使得在组装结果前就知道结果长度，所以免去了集合类的使用。
代码实现
class Solution {
public int[][] insert(int[][] intervals, int[] newInterval) {
int index=0, start=newInterval[0], end=newInterval[1], count=0;
if (intervals.length > 0) {
if (intervals[0][0] > newInterval[1]) {
index = 0;
} else if (intervals[intervals.length-1][1] < newInterval[0]) {
index = intervals.length;
} else {
boolean flag = false;
for (int i=0; i<intervals.length; i++) {
if (flag) {
if (intervals[i][0] > end) {
break;
}
end = Math.max(end, intervals[i][1]);
count++;
} else {
if (intervals[i][0] > newInterval[1]) {
index = i;
break;
}
if (intervals[i][1] >= newInterval[0]) {
flag = true;
index = i;
start = Math.min(newInterval[0], intervals[i][0]);
end = Math.max(newInterval[1], intervals[i][1]);
count++;
}
}
}
}
}
int[][] result = new int[intervals.length-count+1][2];
for (int i=0; i<index; i++) {
result[i] = intervals[i];
}
result[index][0] = start;
result[index][1] = end;
for (int i=index+count; i<intervals.length; i++) {
result[i-count+1] = intervals[i];
}
return result;
}
}
