【LeetCode】插入区间

发布于:2024-07-15 ⋅ 阅读:(110) ⋅ 点赞:(0)


一、题目

给你一个 无重叠的 ,按照区间起始端点排序的区间列表 intervals,其中 intervals[i] = [starti, endi] 表示第 i 个区间的开始和结束,并且 intervals 按照 starti 升序排列。同样给定一个区间 newInterval = [start, end] 表示另一个区间的开始和结束。

在 intervals 中插入区间 newInterval,使得 intervals 依然按照 starti 升序排列,且区间之间不重叠(如果有必要的话,可以合并区间)。

返回插入之后的 intervals。

注意 你不需要原地修改 intervals。你可以创建一个新数组然后返回它。

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

提示:

0 <= intervals.length <= 104
intervals[i].length == 2
0 <= starti <= endi <= 105
intervals 根据 starti 按 升序 排列
newInterval.length == 2
0 <= start <= end <= 105


二、解法

分情况模拟就好了,可以遍历intervals,同时与newInterval做比较,判断将谁插入到res
在插入的时候,判断是否需要合并区间


完整代码

class Solution:
    def insert(self, intervals: List[List[int]], newInterval: List[int]) -> List[List[int]]:
        n = len(intervals)
        res = []
        i = 0
        used = False
        while i < n or not used:
            cur = None
            if not used and (i == n or intervals[i][0] > newInterval[0]):
                cur = newInterval
                used = True
            else:
                cur = intervals[i]
                i += 1
            if not res or res[-1][1] < cur[0]:
                res.append(cur)
            else:
                res[-1][1] = max(res[-1][1], cur[1])
        return res


网站公告

今日签到

点亮在社区的每一天
去签到