1. 程式人生 > >Google面試題專題8 - leetcode57. Insert Interval

Google面試題專題8 - leetcode57. Insert Interval

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