python-leetcode57-区间合并插入区间

Posted 2021-03-08 西西嘛呦

问题描述：

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

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

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

 

有了之前leetcode56的思路，这就简单了，直接先将要插入的区间加入到intervals中，后面代码都是一样的。

class Solution:
    def insert(self, intervals: List[List[int]], newInterval: List[int]) -> List[List[int]]:
        res = []
        intervals.append(newInterval)
        intervals.sort()
        for i in intervals:
            if not res or res[-1][1]<i[0]:
                res.append(i)
            else:
                res[-1][1]=max(res[-1][1],i[1])
        return res

结果：

这是我第二次运行的结果，记一次时间消耗为156ms，击败7%，不知道它的运行机制是怎样的，还是根据自己电脑有关。

再仔细看下题目，说了intervals是按区间端点进行排序的，因此，可以利用二分查找法查找该区间插入的位置。

class Solution:
    def insert(self, intervals: List[List[int]], newInterval: List[int]) -> List[List[int]]:
        if not intervals:
            return [newInterval]
        res = []
        ins = self.helper(intervals,newInterval[0])
        intervals.insert(ins,newInterval)
        for i in intervals:
            if not res or res[-1][1]<i[0]:
                res.append(i)
            else:
                res[-1][1]=max(res[-1][1],i[1])
        return res
    def helper(self,intervals,val):
        l=0
        r=len(intervals)-1
        if intervals[-1][0]<val:
            return r+1
        if intervals[0][0]>val:
            return l
        while l<r:
            mid=(l+r)//2
            if intervals[mid][0]>val:
                r=mid
            else:
                l=mid+1
        return l

结果：

注意要考虑特殊情况，当插入的区间端点大于被插入区间端点的最大值时，要返回len(intervals) ，即插入到被插入区间最后面。