LeetCode 57. 插入区间 Python

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

class Solution:
    def insert(self, intervals: List[List[int]], newInterval: List[int]) -> List[List[int]]:
        n = len(intervals)
        i = 0
        # 从左往右找到第一个重叠区域intervals[i]
        while i < n and intervals[i][0] < newInterval[0]:
            if intervals[i][1] >= newInterval[0]:
                break
            i += 1
       
        if i == n:
            return intervals + [newInterval]

        j = n - 1
        # 从右往左找到最后一个重叠区域intervals[j]
        while j >= 0 and intervals[j][0] > newInterval[1]:
            j -= 1
        
        if j == -1:
            return [newInterval] + intervals
        
        # 合并i到j，和newInterval这些重叠区间
        merge = [min(intervals[i][0], newInterval[0]), max(intervals[j][1], newInterval[1])]
        return intervals[:i] + [merge] + intervals[j + 1:]