LeetCode Merge Intervals

LeetCode解题之Merge Intervals

原题

给出多个数据区段，把首尾相连的数据段合并。

注意点：

• 所给的数据段是乱序的

例子：

输入: intervals = [1,3],[2,6],[8,10],[15,18]

输出: [1,6],[8,10],[15,18]

解题思路

先把所有的数据段按照起始位置(start)排序，这样可以使可能相连的数据段到放到相邻的位置。遍历数据段，并与结果集中最后一个数据段比较能否合并，如果能合并就合并，否则加入结果集。

AC源码

# Definition for an interval.
class Interval(object):
    def __init__(self, s=0, e=0):
        self.start = s
        self.end = e

    # To print the result
    def __str__(self):
        return "[" + str(self.start) + "," + str(self.end) + "]"


class Solution(object):
    def merge(self, intervals):
        """ :type intervals: List[Interval] :rtype: List[Interval] """
        result = []
        if not intervals:
            return result
        intervals.sort(key=lambda x: x.start)
        result.append(intervals[0])
        for interval in intervals[1:]:
            prev = result[-1]
            if prev.end >= interval.start:
                prev.end = max(prev.end, interval.end)
            else:
                result.append(interval)
        return result


if __name__ == "__main__":
    intervals = Solution().merge([Interval(1, 3), Interval(2, 6), Interval(8, 10), Interval(15, 18)])
    for interval in intervals:
        print(interval)

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。