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來獲得相關源碼。