越简洁的代码往往越高效,越易读。

这题一种简洁的解法是使用一个新的数组res保存结果,扫描intervals,newInterval可能在当前interval的左边,右边,或者相交。

如果是前两种情况,则直接把newInterval放入res,如果是相交,则更新newInterval的左右边界,成为新的newInterval。

bool变量has_put用于标识newInterval是否已经放入res。


/**
 * Definition for an interval.
 * struct Interval {
 *     int start;
 *     int end;
 *     Interval() : start(0), end(0) {}
 *     Interval(int s, int e) : start(s), end(e) {}
 * };
 */
class Solution {
public:
    vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {
		vector<Interval> res;
		bool has_put = false;
		for(int i=0;i<intervals.size();++i){
			if(intervals[i].end<newInterval.start)
				res.push_back(intervals[i]);
			else if( intervals[i].start>newInterval.end){
				if(!has_put)
					res.push_back(newInterval);
				has_put = true;
				res.push_back(intervals[i]);
			}
			else if(intervals[i].start<=newInterval.start && intervals[i].end>=newInterval.end){
				res.push_back(intervals[i]);
				has_put = true;
			}
			else{
				newInterval.start = min(intervals[i].start,newInterval.start);
				newInterval.end = max(intervals[i].end,newInterval.end);
			}
		}
		if(!has_put)
			res.push_back(newInterval);
		return res;
	}
};


Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐