57. Insert Interval (Medium) - TengnanYao/daily_leetcode GitHub Wiki

class Solution(object):
    def insert(self, intervals, newInterval):
        """
        :type intervals: List[List[int]]
        :type newInterval: List[int]
        :rtype: List[List[int]]
        """
        a0, a1 = newInterval
        n = len(intervals)
        i = 0
        while i < n and a0 > intervals[i][1]:
            i += 1
        j = i
        while j < n and a0 <= intervals[j][1] and intervals[j][0] <= a1:
            a0, a1 = min(a0, intervals[j][0]), max(a1, intervals[j][1])
            j += 1
        return intervals[ : i] + [[a0, a1]] + intervals[j : ]