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

时间:2022-07-24
本文章向大家介绍【python-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] 重叠。

有了之前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) ,即插入到被插入区间最后面。