leetcode 57. 插入区间

题目描述

给出一个无重叠的 ，按照区间起始端点排序的区间列表。

在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。

示例 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] 重叠。

解题思路

首先判断intervals的元素与newInterval是否存在交集，如果存在的话，合并交集区间，如果不存在，确定插入位置。

代码（python）

class Solution(object):
    def intersection(self,interval1,interval2):
        temp=None
        if interval1[0]>interval2[0]:
            temp=interval1
            interval1=interval2
            interval2=temp
        if interval1[1]>=interval2[0]:
            return True
        return False
            
    def insert(self, intervals, newInterval):
        """
        :type intervals: List[List[int]]
        :type newInterval: List[int]
        :rtype: List[List[int]]
        """
        if len(intervals)==0:
            intervals.append(newInterval)
            return intervals
        indexIntersection=[]
        startIndex=-1
        for index,interval in enumerate(intervals):
            if self.intersection(interval,newInterval):
                indexIntersection.append(index)
            if interval[0]>newInterval[0] and startIndex==-1:
                startIndex=index
        if len(indexIntersection)>0:
            left=[intervals[index][0] for index in indexIntersection]
            left.append(newInterval[0])
            right=[intervals[index][1] for index in indexIntersection]
            right.append(newInterval[1])
            left_min=min(left)
            right_max=max(right)
            intervals[indexIntersection[0]]=[left_min,right_max]
            indexIntersection.reverse()
            for index in indexIntersection[:-1]:
                del(intervals[index])
        elif len(indexIntersection)==0:
            if startIndex==-1:
                if intervals[0][0]>newInterval[1]:
                    intervals.insert(0,newInterval)
                else:
                    intervals.append(newInterval)
            else:
                intervals.insert(startIndex,newInterval)
        return intervals