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) 来获得相关源码。