leetcode

合并区间

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] 可被视为重叠区间。

思路:

  1. 将intervals按每一个元素的start进行升序排列。
  2. 此时后一个值的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的值。接下来继续循环判断。
  3. 循环结束再往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
上一篇 下一篇

猜你喜欢

热点阅读