leetcode57. 插入区间

题目：

给出一个无重叠的 ，按照区间起始端点排序的区间列表。
在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。

示例 1:

输入: intervals = [[1,3],[6,9]], newInterval = [2,5]
输出: [[1,5],[6,9]]

示例 2:

输入: intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
输出: [[1,2],[3,10],[12,16]]
解释: 这是因为新的区间 [4,8] 与 [3,5],[6,7],[8,10] 重叠。

思路：

• 遍历原数组，如果当前区间比要插入的区间小，直接把当前区间放到结果数组中；
• 如果找到了要插入的位置，插入之前先检查是否重合，如果重合则需要先合并后插入。
• 依次把剩余的数组放到结果数组中，如果有重合就合并。

代码：

class Solution(object):
    def insert(self, intervals, newInterval):
        """
        :type intervals: List[List[int]]
        :type newInterval: List[int]
        :rtype: List[List[int]]
        """
        if not intervals:
            return [newInterval]
        res = []
        pos = 0
        for i in range(len(intervals)):
            if intervals[i][0] < newInterval[0]:
                res.append(intervals[i])
                pos = i
                continue
            if intervals[i][0] >= newInterval[0]:
                pos = i
                break

        if res:
            temp = res[-1]
            if newInterval[0] <= temp[1]:
                res.pop()
                res.append([temp[0], max(newInterval[1], temp[1])])
            else:
                res.append(newInterval)
        else:
            res.append(newInterval)

        for j in range(pos, len(intervals)):
            if res:
                temp = res[-1]
                if intervals[j][0] <= temp[1]:
                    res.pop()
                    res.append([temp[0], max(intervals[j][1], temp[1])])
                else:
                    res.append(intervals[j])
            else:
                res.append(intervals[j])
        return res