Leetcode 57. 插入区间

题目

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

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

示例 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] 重叠。

解答

与上一题合并区间类似。

具体步骤就是在加入新的区间后，对新的集合进行合并区间即可。

合并区间请看上一篇博客：https://blog.youkuaiyun.com/LetJava/article/details/95617026

代码

class Solution {
    public int[][] insert(int[][] intervals, int[] newInterval) {
        List<int[]> arr = new ArrayList<>();
        for(int i = 0; i < intervals.length; i ++) arr.add(intervals[i]);
        
        // 找到插入位置，并插入新区间
        int index = 0;
        while(index < intervals.length && newInterval[0] >= intervals[index][0]) index ++;
        arr.add(index, newInterval);
        
        // 合并区间
        List<int[]> res = new ArrayList<>();
        for(int i = 0; i < arr.size(); i ++) {
            int start = arr.get(i)[0];
            int end = arr.get(i)[1];
            while(i < arr.size() - 1 && arr.get(i + 1)[0] <= end) {
                end = Math.max(end, arr.get(i + 1)[1]);
                i ++;
            }
            
            res.add(new int[]{start, end});
        }
        
        return res.toArray(new int[0][]);                
    }
    
}

结果