Python 的 LeetCode 之旅(4):57 Insert Interval

原题链接:https://leetcode.com/problems/insert-interval/description/
题目描述:

Given a set of non-overlapping intervals, insert a new interval into the intervals (merge if necessary).

You may assume that the intervals were initially sorted according to their start times.

Example 1:

Input: intervals = [[1,3],[6,9]], newInterval = [2,5]
Output: [[1,5],[6,9]]

Example 2:

Input: intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
Output: [[1,2],[3,10],[12,16]]
Explanation: Because the new interval [4,8] overlaps with [3,5],[6,7],[8,10].

题目大意:

给出一系列已经排序好的区间,然后加入一个新的区间,要求合并这些区间

解题思路:

从前向后遍历区间,找到第一个受到新加入的区间影响的区间,对于该区间以前的区间全部直接加入结果,然后接着向后遍历找到受到新加入区间影响的最后一个区间,该区间之后的区间也是直接加入结果,而中间的区间合并成一个新的区间即可。

代码:

class Solution:
    def insert(self, intervals, newInterval):
        start = newInterval.start
        end = newInterval.end
        result = []
        i = 0
        while i < len(intervals):
            if start <= intervals[i].end:
                if end < intervals[i].start:
                    break
                start = min(start, intervals[i].start)
                end = max(end, intervals[i].end)
            else:
                result.append(intervals[i])
            i += 1
        result.append(Interval(start, end))
        result += intervals[i:]
        return result
        

运行结果:

猜你喜欢

转载自blog.csdn.net/qq_36185481/article/details/80286943