【每天一道算法题11】合并区间

题目描述

给出一个区间的集合，请合并所有重叠的区间。

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

示例 2:
输入: [[1,4],[4,5]]
输出: [[1,5]]
解释: 区间 [1,4] 和 [4,5] 可被视为重叠区间

解法：先排序后合并

class Solution:
    def merge(self, intervals: List[List[int]]) -> List[List[int]]:
        if not intervals:
            return []
        # 先对区间按照左边界进行排序
        intervals = sorted(intervals, key=lambda x:x[0])
        # 排序后的区间列表，相邻两个区间A[x1, y1] B[x2, y2]只存在以下三种情况
        # 1. A和B不相交，此时 y1 < x2
        # 2. A包含B, 此时 y1 >= y2
        # 3. A和B相交, 此时 x2 < y1 < y2
        res = [intervals[0]]
        for x in intervals:
            # 不相交的时，直接把x压入结果集
            if x[0] > res[-1][1]:
                res.append(x)
            else:
                # 相交时，合并后重置res[-1]，至于包含的情况，可以直接跳过
                if x[1] > res[-1][1]:
                    res[-1] = [res[-1][0], x[1]]
        return res