组队(贪心算法)

#include <iostream>
#include <vector>
#include <algorithm> //算法头文件
using namespace std;
/*
链接：https://www.nowcoder.com/questionTerminal/248ccf8b479c49a98790db17251e39bb
来源：牛客网

牛牛举办了一次编程比赛,参加比赛的有3*n个选手,每个选手都有一个水平值a_i.现在要将这些选手进行组队,一共组成n个队伍,即每个队伍3人.牛牛发现队伍的水平值等于该队伍队员中第二高水平值。
例如:
一个队伍三个队员的水平值分别是3,3,3.那么队伍的水平值是3
一个队伍三个队员的水平值分别是3,2,3.那么队伍的水平值是3
一个队伍三个队员的水平值分别是1,5,2.那么队伍的水平值是2
为了让比赛更有看点,牛牛想安排队伍使所有队伍的水平值总和最大。
如样例所示:
如果牛牛把6个队员划分到两个队伍
如果方案为:
team1:{1,2,5}, team2:{5,5,8}, 这时候水平值总和为7.
而如果方案为:
team1:{2,5,8}, team2:{1,5,5}, 这时候水平值总和为10.
没有比总和为10更大的方案,所以输出10.
*/
int main()
{
	int n;
	long long int a_i;
	while (1)
	{
		cin >> n; //输入组数
		if (n >= 1 && n < 10000) {
			break;
		}
	}
	vector<int> a_n;
	while (a_n.size() < 3 * n) 
	{
		cin >> a_i; //输入每个人的实力值
		a_n.push_back(a_i);
	}
	sort(a_n.begin(), a_n.end()); //对每个人的实力进行从小到大的排序
	long long int max = 0;
	/*
	为了使所有组加起来的实力值达到最大,
	我们让前n个人作为每组的实力最弱的人,
	然后对后面2n个人进行分组,
	得到最大值的分组方法是: 第n个人和n+1个人分到一组
	这样每一组的第二实力为 n, n+2, n+4,.....3n-1; 
	这就是我们要求的势力最大值
	*/
	for ( n; n < a_n.size(); n += 2)
	{
		max += a_n[n]; //将每组第二的实力值加起来
	}
	cout << max << endl;
	return 0;
}