【Leetcode】no.56 合并区间

欢迎各位来关注我的个人网站：https://tyyuan110.com/
我会在上面定期更新相关技术博客（前端、测开、面经等），谢谢！

Question

以数组 intervals 表示若干个区间的集合，其中单个区间为 intervals[i] = [start, end] 。
请你合并所有重叠的区间，并返回 一个不重叠的区间数组，该数组需恰好覆盖输入中的所有区间 。

输入：intervals = [[1,3],[2,6],[8,10],[15,18]]
输出：[[1,6],[8,10],[15,18]]
解释：区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].
示例 2：

输入：intervals = [[1,4],[4,5]]
输出：[[1,5]]
解释：区间 [1,4] 和 [4,5] 可被视为重叠区间。

# 提示：

1 <= intervals.length <= 104
intervals[i].length == 2
0 <= starti <= endi <= 104

Solution

假设区间集合如下

intervals = [[5, 6], [2, 6], [1, 3], [8, 10], [15, 18]]

对于其中的一个区间我们叫做 pair ，我们首先根据 pair[0] 来对整个 intervals 进行排序。
排序后得到：

intervals = [[1, 3], [2, 6], [5, 6], [8, 10], [15, 18]]

然后我们通过一次遍历来合并区间。
首先取出第一个 pair1 ，在取出第二个 pair2 来进行比较，每次比较流程如下：

pair1 的最大值是否大于 pair2 的最小值：

1. 大于：说明两区间重叠。接下来两区间的最大值即为合并后区间的最大值；
2. 不大于：说明两区间不重叠。将 pair1 加入到存放答案的列表中

最后代码如下所示：

from typing import List


def merge(intervals: List[List[int]]) -> List[List[int]]:
    if len(intervals) == 1:
        return intervals

    intervals.sort(key=lambda x: x[0])

    ans = []
    tmp = intervals[0]
    for pair in intervals[1:]:
        if tmp[1] >= pair[0]:
            tmp[1] = max(tmp[1], pair[1])
        else:
            ans.append(tmp)
            tmp = pair
    ans.append(tmp) # the last tmp should be pushed into the list
    return ans

Result

• 执行用时：44 ms, 在所有 Python3 提交中击败了91.92%的用户
• 内存消耗：17.9 MB, 在所有 Python3 提交中击败了74.13%的用户