Leetcode题库57. 插入区间(python实现)

思路

第一步：
将待插入数组插入到合适的位置
例：
intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]]
newInterval = [4,8]
那么新的intervals
intervals = [[1,2],[3,5],[4,8],[6,7],[8,10],[12,16]]

第二步：
此时，问题就变成了合并intervals区间
与leetcode56. 合并区间 题目相同
于是就只需要再调用该题目中的函数返回结果即可
官网题目链接：链接
题解链接：链接

代码

def merge(intervals: List[List[int]]) -> List[List[int]]:
    flag=0
    for i in intervals:
        i.insert(0,flag)
        flag=flag+1

    sorted0_intervals = sorted(intervals, key = lambda x:x[1])
    sorted1_intervals = sorted(intervals, key = lambda x:x[2])
    t = []
    i = 0
    while i < len(sorted0_intervals):
        t.append(sorted0_intervals[i])
        for j in range(len(sorted0_intervals)-1,-1,-1):
            if sorted0_intervals[i] == sorted1_intervals[j]:
                i = j + 1
                break
    i = 0
    while i < len(t)-1:
        if t[i][2] >= t[i+1][1]:
            t[i+1][1] = t[i][1]
            del t[i]
        else:
            i = i + 1
    
    ret = []
    for i in t:
        ret.append([i[1],i[2]])

    return ret

class Solution:
    def insert(self, intervals: List[List[int]], newInterval: List[int]) -> List[List[int]]:
        flag=0
        for i in range(len(intervals)):
            if(intervals[i][0]<=newInterval[0]<=intervals[i][1]):
                flag=i+1
                break
        intervals.insert(flag , newInterval)
        intervals=merge(intervals=intervals)
        return intervals