leetcode上的原题:
Given a collection of intervals, merge all overlapping intervals.
For example,
Given [1,3],[2,6],[8,10],[15,18]
,
return [1,6],[8,10],[15,18]
.
题意应该好理解,比如 1,3 与2,6都是被标记的范围,要求改为用1,6来表示。
这个是用来表示间隙的结构体。
/**
* Definition for an interval.
* struct Interval {
* int start;
* int end;
* Interval() : start(0), end(0) {}
* Interval(int s, int e) : start(s), end(e) {}
* };
*/
解决思路:
首先排序,把Interval们按start的升序排列,这样可以合并的Interval肯定就挨在一起了,然后找出相邻可以合并的Interval的最大end,就能合并了。
感觉没什么好说的,上代码吧。
bool operator < (Interval a, Interval b)
{
return a.start < b.start;
}
class Solution
{
public:
vector<Interval> merge(vector<Interval> &intervals)
{
if (intervals.size() < 2)
return intervals;
sort(intervals.begin(), intervals.end());
vector<Interval>res;
for (auto i = intervals.begin(); i != intervals.end() ;)
{
res.push_back(*i);
for (i++; i != intervals.end() && res.back().end >= i->start; i++)
res.back().end = max(i->end, res.back().end);
}
return res;
}
};
这里为了用库的快排,重载了一个<,也可以 写一个比较函数cmp,原理一样的,比较start就行。
判断是否可以合并是用后一个interval的start与当前interval的end比较,可以看出两个interval是否交叉。
注意:第一个循环不能把i++,因为i已经在子循环中达到了一个用于合并的interval的头部。
ac过后发现用了24ms,本来自我感觉还行的,结果看到竟然没有达到平均水平,于是干脆把i变成数组下标。快了一点,21ms完成。达到了平均水准,但是好多大神好像在个位数ms就搞定了,估计还有什么可以优化的地方……