您的位置:首页 > 其它

[leetcode]Insert Interval

2014-01-04 13:22 393 查看
偷懒,直接用了Merge Interval的代码。。。

吧那个Interval加入到前面的vector,然后就和Merge Interval一样的了。。。TAT

..几天前才写过的代码,结果都又是各种问题。。。唉。。。代码能力啊。。。。

/**
* Definition for an interval.
* struct Interval {
*     int start;
*     int end;
*     Interval() : start(0), end(0) {}
*     Interval(int s, int e) : start(s), end(e) {}
* };
*/
bool cmp(const Interval& a , const Interval& b){
return a.start < b.start;
}
class Solution {
public:
vector<Interval> insert(vector<Interval> &intervals, Interval newInterval) {
intervals.push_back(newInterval);
sort(intervals.begin(),intervals.end() , cmp);

vector<Interval> ans;
int size = intervals.size();
if(size == 1) return intervals;
int start = intervals[0].start;
int end = intervals[0].end;
for(int i = 1 ; i < size ;){
while(i < size  && intervals[i].start <= end)
{
if(intervals[i].end > end) end = intervals[i].end;
i++;
}

ans.push_back(Interval(start , end));
start = intervals[i].start;
end = intervals[i].end;

}
//    ans.push_back(Interval(start , end));
return ans;
}
};


--update 2014/01/25

之前的做法毕竟太屌丝。。。

正常做法应该是

找到start的位置

因为是按start排序好了的,所以我们的start,必然是>前面所有的end

start = min(start , interals[i].start) //如果插入的是最后的,那么不需要这个判断

然后呢,从后面的来看。。如果后面的start《=我们的end,那么合并

插入这个新的

/**
* 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) {
int start = newInterval.start;
int end = newInterval.end;
int i = 0;
vector<Interval> ans;
for( ; i < intervals.size() && start > intervals[i].end ; i++) {
ans.push_back(intervals[i]);
}
//found i , not the last
if (i < intervals.size())
start  = min(start, intervals[i].start);
for(; i < intervals.size() && end >= intervals[i].start ; i++) {
end = max(end , intervals[i].end);
}
ans.push_back(Interval(start,end));
ans.insert(ans.end() , intervals.begin() + i , intervals.end());
return ans;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: