LeetCode——区间合并

区间合并

给出一个区间的集合，请合并所有重叠的区间。

示例 1:

输入: [[1,3],[2,6],[8,10],[15,18]]
输出: [[1,6],[8,10],[15,18]]
解释: 区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].

示例 2:

输入: [[1,4],[4,5]]
输出: [[1,5]]
解释: 区间 [1,4] 和 [4,5] 可被视为重叠区间。

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

class point
{
public:
	int a;
	int b;
	point(int x, int y) :a(x), b(y) {}
};

vector<point> merge(vector<point> &pvec)
{
	if (pvec.empty())
		return pvec;
	vector<point> ret;
	sort(pvec.begin(), pvec.end(), [](point A, point B)->bool {return (A.a < B.a); });
	int size = pvec.size();
	ret.push_back(pvec[0]);
	for (int i = 1; i < size;i++)
	{
		if (ret.back().b >= pvec[i].a)
		{
			ret.back().b = max(ret.back().b, pvec[i].b);
		}
		else
		{
			ret.push_back(pvec[i]);
		}
	}
	return ret;
}

int main()
{
	vector<point> pvec = { { 2, 6 }, { 1, 3 }, { 8, 10 }, { 15, 18 } };
	vector<point> ret = merge(pvec);
	for (auto m : ret)
		cout << "[" << m.a << "," << m.b << "]" << endl;
	return 0;
}