合并区间
2018-09-12 本文已影响9人
momo1023
给出一个区间的集合,请合并所有重叠的区间。
输入: [[1,3],[2,6],[8,10],[15,18]]
输出: [[1,6],[8,10],[15,18]]
解释: 区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].
输入: [[1,4],[4,5]]
输出: [[1,5]]
解释: 区间 [1,4] 和 [4,5] 可被视为重叠区间。
思路:
- 将intervals按每一个元素的start进行升序排列。
- 此时后一个值的start一定在前一个值的start后(或相等)。这个时候只要判断后一个的start是否比前一个的end大。这里我设置了两个指针l和h来表示区间的起始值和终点,列表res作为结果。判断:如果 intervals[i].start <= intervals[i-1].end, 那么l保持不变,h为max(intervals[i].end, intervals[i-1].end)。否则,往列表res添加[l,h],更新l和h的值。接下来继续循环判断。
- 循环结束再往res添加[l,h]。
# Definition for an interval.
# class Interval:
# def __init__(self, s=0, e=0):
# self.start = s
# self.end = e
class Solution:
def merge(self, intervals):
"""
:type intervals: List[Interval]
:rtype: List[Interval]
"""
if len(intervals) <= 1:
return intervals
res = []
intervals = sorted(intervals, key=lambda x : x.start)
l = intervals[0].start
h = intervals[0].end
for i in intervals:
if i.start <= h:
h = max(h, i.end)
else:
res.append([l, h])
l = i.start
h = i.end
res.append([l, h])
return res