給出多個不重合的數據區段,現在插入一個數據區段,有重合的區段要進行合並。
注意點:
所給的區段已經按照起始位置進行排序例子:
輸入: intervals = [2,6],[8,10],[15,18], newInterval = [13,16]
輸出: [2,6],[8,10],[13,18]
最簡單的方式就是復用 Merge Intervals 的方法,只需先將新的數據區段加入集合即可,但這樣效率不高。既然原來的數據段是有序且不重合的,那麼我們只需要找到哪些數據段與新的數據段重合,把這些數據段合並,並加上它左右的數據段即可。
# 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來獲得相關源碼。