LeetCode 57. 插入区间(python、c++)

题目描述

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

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

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

题解

将该点加入到原来的数组中
1.首先将所有区间按照 start 排序,start 相同的按照 end 排序。
2.从头开始遍历所有区间,并求他们的并集。定义 cur 表示当前一些区间的并集。
3.若发现当前遍历到的区间 start 严格大于 cur 的 end,则当前并集需要加入到答案数组中,且更新区间并集 cur 为当前区间。
4.若发现当前遍历到的区间 end 严格大于 cur 的 end,则更新 cur 的 end 为当前区间的 end。
时间复杂度
排序的时间复杂度为 O(nlogn),排序后只需要线性扫描,故总时间复杂度为 O(nlogn)。

c++代码:

class Solution {
    
    
public:
    vector<vector<int>> insert(vector<vector<int>>& intervals, vector<int>& newInterval) {
    
    
        intervals.push_back(newInterval);
        sort(intervals.begin(), intervals.end());
        int n = intervals.size();
        vector<vector<int>> ans;
        for(int i = 0; i < n; i++){
    
    
            int l = intervals[i][0], r = intervals[i][1];
            if(!ans.size() || ans.back()[1] < l)
                ans.push_back({
    
    l, r});
            else
                ans.back()[1] = max(ans.back()[1], r);
        }
        return ans;
    }
};

python代码:

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

猜你喜欢

转载自blog.csdn.net/qq_43328040/article/details/109063348