LeetCode No57. 插入区间

题目

给你一个 无重叠的 ，按照区间起始端点排序的区间列表。

在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。

示例 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

思路

模拟题，维护一个区间[low, high]，遍历原数组，如果当前区间与[low, high]无交集，则直接添加到新数组中，如果有交集则重新维护[low, high]。

AC代码

点击查看代码

class Solution {
    public int[][] insert(int[][] intervals, int[] newInterval) {
        int[][] res = new int[intervals.length+1][2];
        int len = intervals.length;
        int index = 0;
        if( len==0 ) {
            res[index++] = newInterval;
            return Arrays.copyOf(res, index);
        }
        boolean flag = false;
        int low = newInterval[0];
        int high = newInterval[1];
        for(int i=0; i<len; i++ ) {
            if (intervals[i][0] > high) {
                if( !flag ) {
                    res[index++] = new int[]{low, high};
                    flag = true;
                }
                res[index++] = intervals[i];
            } else if( intervals[i][1]<low) {
                res[index++] = intervals[i];
            } else {
                low = Math.min(low, intervals[i][0]);
                high = Math.max(high, intervals[i][1]);
            }
        }
        if( !flag ) {
            res[index++] = new int[]{low, high};
        }
        return Arrays.copyOf(res, index);
    }
}