[Leetcode]合并区间
题目描述
给出一个区间的集合,请合并所有重叠的区间。
示例 1:
输入: 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] 可被视为重叠区间。
注意:输入类型已于2019年4月15日更改。 请重置默认代码定义以获取新方法签名。
提示:
intervals[i][0] <= intervals[i][1]
解题思路
- 先对数组进行排序,这样就可以保证 tmp1 与 tmp2 做合并时,tmp1[0] < tmp2[0]
- 然后,对于 tmp1 与 tmp2 有三种情况:
-
交叉
-
包含
-
不相交
-
- 那么,比较 tmp1[1] 与 tmp2[1],可以判断是 相交 还是 包含,进行区间合并;比较 tmp1[1] 与 tmp2[0] 可以判断是否是不相交
实现代码
class Solution:
def merge(self, intervals: List[List[int]]) -> List[List[int]]:
intervals.sort(key=lambda x: x[0])
merged = []
for interval in intervals:
# 如果列表为空,或者当前区间与上一区间不重合,直接添加
if not merged or merged[-1][1] < interval[0]:
merged.append(interval)
else:
# 否则的话,我们就可以与上一区间进行合并
merged[-1][1] = max(merged[-1][1], interval[1])
return merged