【LeetCode-贪心】插入区间

题目描述

给出一个无重叠的 ,按照区间起始端点排序的区间列表。
在列表中插入一个新的区间,你需要确保列表中的区间仍然有序且不重叠(如果有必要的话,可以合并区间)。
示例:

输入: intervals = [[1,3],[6,9]], newInterval = [2,5]
输出: [[1,5],[6,9]]

输入: 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] 重叠。

题目链接: https://leetcode-cn.com/problems/insert-interval/

思路

遍历给定的区间列表,如果当前区间右端点 intervals[i][1]>newInterval[0],说明要插入的区间和当前区间有交集;如果该条件不满足,则将当前区间放进答案数组 ans 中,并在当前位置中断;

将新区间 newInterval 放入 ans 中;

从中断的位置 i 开始继续遍历 intervals:

  • 假设 ans 数组的最后一个区间为 back;
  • 如果 back[1]<intervals[i][0],则说明当前区间以及之后的区间都不会和 newInterval 有交集,可以直接放入 ans 中;
  • 否则的话,更新 back 的两个端点,具体见代码;

代码如下:

class Solution {
public:
    vector<vector<int>> insert(vector<vector<int>>& intervals, vector<int>& newInterval) {
        if(intervals.empty()) return {newInterval};

        vector<vector<int>> ans;
        int i=0;
        for(; i<intervals.size(); i++){
            if(intervals[i][1]<newInterval[0]) ans.push_back(intervals[i]);
            else break;
        }

        ans.push_back(newInterval); // 别忘了这一步
        for(; i<intervals.size(); i++){
            if(ans.back()[1]<intervals[i][0]) ans.push_back(intervals[i]);
            else{
                ans.back()[0] = min(ans.back()[0], intervals[i][0]);  // 左端点用 min
                ans.back()[1] = max(ans.back()[1], intervals[i][1]);  // 右端点用 max
            }
        }
        return ans;
    }
};
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

参考

1、https://leetcode-cn.com/problems/insert-interval/solution/cyi-ci-bian-li-zhi-jie-fa-jiao-hao-li-jie-by-wo-sh/323281

posted @ 2020-06-14 22:59  Flix  阅读(193)  评论(0编辑  收藏  举报