leetcode-组队竞赛-69

Posted 2021-06-20 天津 唐秙

题目要求
  牛牛举办了一次编程比赛,参加比赛的有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.

思路
  输入队伍数量n，我们预留3*n个空间的数组，我们将值写到数组当中，然后将数组进行排序，因为平均值是3个数中的中位数，所以第一个数和最后一个数分别是数组中最大和最小的，为了平均值最大，中位数最好是数组中第二大的数字，最后将所有的中位数相加。

代码实现

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
	int n;
	while (cin >> n)
	{
		long long val = 0;
		vector<int> arr;
		arr.resize(3 * n);
		for (int i = 0; i < 3*n; i++)
		{
			cin >> arr[i];
		}
		sort(arr.begin(), arr.end());

		for (int i = 0; i < n; i++)
		{
			val = val + arr[arr.size() - (2 * (i + 1))];
		}
		cout << val << endl;
	}
	return 0;
}