题目

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
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
/**
* 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:
// 定义排序函数 按照每个片段的start的大小从小到大排序。
static bool cmp(Interval a,Interval b)
{
return a.start<b.start;
}
vector<Interval> merge(vector<Interval>& intervals)
{
vector<Interval> res;
if(intervals.size()==0)
{
return res;
}
sort(intervals.begin(),intervals.end(),cmp);
res.push_back(intervals[0]);
for(int i=1;i<intervals.size();i++)
{
if(res.back().end>=intervals[i].start)
{

Interval temp(res.back().start,max(res.back().end,intervals[i].end));
res.pop_back();
res.push_back(temp);//注意pop和push的顺序,第一次这里的顺序浪费了一些时间。
}
else
{
res.push_back(intervals[i]);
}
}
return res;
}
};

思路

先将所有的片段按照start的大小进行排序,考虑 13 24 56 这种情况,先将第一个片段放入result,然后比较第二个片段的开始大小和第一个片段的结束大小的关系,若第一个结束的大小小于第二个开始的大小,直接将第二个片段放入result即可。若第一个结束的大小大于第二个结束的大小,则去求应该把哪个的结束作为这个片段的结束,就像 13 24 比时,比较3 和4 哪个作为结束,所以把上一个片段退出来,重新生成一个新的片段,然后加入result中。