LeetCode解題之Insert Interval

原題

給出多個不重合的數據區段，如今插入一個數據區段。有重合的區段要進行合並。

註意點：

• 所給的區段已經依照起始位置進行排序

樣例：

輸入: intervals = [2,6],[8,10],[15,18]。 newInterval = [13,16]

輸出: [2,6],[8,10],[13,18]

解題思路

最簡單的方式就是復用 Merge Intervals 的方法，僅僅需先將新的數據區段增加集合就可以。但這樣效率不高。既然原來的數據段是有序且不重合的，那麽我們僅僅須要找到哪些數據段與新的數據段重合，把這些數據段合並。並加上它左右的數據段就可以。

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 insert(self, intervals, newInterval)
 
:
        start, end = newInterval.start, newInterval.end
        left = list(filter(lambda x: x.end < start, intervals))
        right = list(filter(lambda x: x.start > end, intervals))
        if len(left) + len(right) != len(intervals):
            start = min(start, intervals[len(left)].start)
            end = max(end, intervals[-len(right) - 1
 
].end)

        return left + [Interval(start, end)] + right


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

歡迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 來獲得相關源代碼。

LeetCode Insert Interval