30. 插入区间

描述

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

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

样例

样例 1：

输入:
(2, 5) into [(1,2), (5,9)]
输出:
[(1,9)]

样例 2：

输入:
(3, 4) into [(1,2), (5,9)]
输出:
[(1,2), (3,4), (5,9)]

题解：

class Solution {
public:
  /**
   * Insert newInterval into intervals.
   * @param intervals: Sorted interval list.
   * @param newInterval: new interval.
   * @return: A new interval list.
   */
  vector<Interval> insert(vector<Interval>& intervals, Interval newInterval) {
    // write your code here
    vector<Interval> ans;
    int st;
    for (st = 0; st < intervals.size() && intervals[st].start < newInterval.start; st++) { 
      ans.push_back(intervals[st]);
    } 
      
    if (!ans.empty() && ans.back().end >= newInterval.start) { 
      ans.back().end = max(ans.back().end, newInterval.end);
    } else { 
      ans.push_back(newInterval);    
     } 
     
    for (int i = st; i < intervals.size(); i++) { 
      if (ans.back().end >= intervals[i].start) { 
        ans.back().end = max(ans.back().end, intervals[i].end);
      } else { 
        ans.push_back(intervals[i]);
      } 
    } 
    return ans;  
  }
};