Google面試題專題8 - leetcode57. Insert Interval
阿新 • • 發佈:2018-12-04
57. Insert Interval - Mark
題目描述
給定無重疊區間的列表,在列表中插入新區間。
假設初始時區間已經根據start排好序。
例子
Example 1:
Input: intervals = [[1,3],[6,9]], newInterval = [2,5]
Output: [[1,5],[6,9]]
Example 2:
Input: intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
Output: [[1,2],[3,10],[12,16]]
Explanation: Because the new interval [4,8] overlaps with [3,5],[6,7],[8,10].
思想
肯定需要遍歷一遍intervals,所以時間複雜度至少O(n)。
要點:找到哪些區間有可能和newInterval重合,把這些區間合併,並加上它左右的區間即可。
解法
時間複雜度O(n)
# Definition for an interval.
# class Interval(object):
# def __init__(self, s=0, e=0):
# self.start = s
# self.end = e
class Solution(object):
def insert(self, intervals, newInterval):
"""
:type intervals: List[Interval]
:type newInterval: Interval
:rtype: List[Interval]
"""
start, end = newInterval.start, newInterval.end
left, right = [], []
for interval in intervals:
if interval.end < start:
left.append(interval)
elif interval.start > end:
right.append(interval)
# overlapping
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