算法作业 - 区间覆盖问题(贪心)

算法主要思想: 第一步将所有点进行排序，以方便计算距离。然后把所有区间间距排序，再连接间距最短的两个线段（间距大小并列的只连接一个），统计这时候有多少条线段，如果大于限制线段数就继续连接区间间距次小的两个区间…直到线段数等于规定最大线段数为止。

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

#define max 201

using std::cin;
using std::cout;
using std::endl;
using std::string;
using std::vector;

struct p
{
	int start;
	int end;
};

int main()
{
	int n, m;
	cin >> n >> m;
	vector<p>points;
	int temp;
	for (auto i = 0; i < n; i++)
	{
		cin >> temp;
		p point{ temp - 1, temp };
		points.push_back(point);
	}
	//对所有点进行排序 方便计算距离
	std::sort(begin(points), end(points), [](p x, p y) {return x.start < y.start;});
	vector<int>distance(points.size() - 1, 0);
	for (auto i = 1; i < points.size(); i++)
	{
		distance[i - 1] = points[i].start - points[i - 1].end;
	}
	//对距离进行排序
	std::sort(begin(distance), end(distance));
	auto length = 0;
	int numLine = n;
	for (auto i = 0; i < distance.size(); i++)
	{
		if (i == 0)
		{
			length = distance[i] + 2;
			if (--numLine == m)
			{
				length -= 2;
				break;
			}
		}
		else
		{
			length += distance[i] + 1;
			if (--numLine == m)
			{
				length += (n - i - 2);
				break;
			}
		}
	}
	cout << length << endl;
 }