Leetcode算法——57、插入区间(insert interval)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/HappyRocking/article/details/84970977

给定一系列不重复的区间,要求将一个新的区间插入到这些集合中(如果有必要,则合并)。

假设这些区间已经根据起始位置排好序了。

示例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]]

思路

本题与Leetcode算法——56、合并区间(merge intervals) 很相似,本题多了一步插入区间。

在插入区间之前,首先要找到需要插入到哪个位置,从左到右扫描,然后比较新区间和数组内区间的起始点的大小即可找到。

然后将新区间插入到数组(插入的过程可能会涉及到与上一个区间的重叠合并),插入之后,剩余的子数组可能会与这个区间有重叠部分,需要依次进行扫描判断。

python实现

def insert(intervals, newInterval):
    """
    :type intervals: List[Interval]
    :type newInterval: Interval
    :rtype: List[Interval]
    """
    result = []
    # 查询新的区间应该插入的地方
    i = 0
    for i in range(len(intervals)):
        if intervals[i][0] >= newInterval[0]:
            break
        result.append(intervals[i])
    
    # 处理剩下的区间
    for cur_inter in [newInterval] + intervals[i:]:
        if result and cur_inter[0] <= result[-1][1]:
            result[-1][1] = max(result[-1][1], cur_inter[1])
        else:
            result.append(cur_inter)
    
    return result

if '__main__' == __name__:
    intervals = []
    newInterval = [4,8]
    print(insert(intervals, newInterval))
            

猜你喜欢

转载自blog.csdn.net/HappyRocking/article/details/84970977