目录
牛客_组队竞赛_贪心
题目解析
C++代码1
C++代码2
Java代码
牛客_组队竞赛_贪心
组队竞赛_牛客笔试题_牛客网 (nowcoder.com)
描述:
牛牛举办了一次编程比赛,参加比赛的有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.
题目解析
小贪心,最高分我们要不到,只能退而求其次拿到倒数第二个人的分数。然后一直递归这个策略。
C++代码1
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
/*
2
5 2 8 5 1 5 -> 1 5 8 ,2 5 5用例说明错了吧
1 2 5 5 5 8 (最小,第二大,第一大)
5 2 8 5 1 5 9 9 9
1 9 9 ,2 8 9 , 5 5 5
*/
#define int long long
signed main()
{
int n = 0;
cin >> n;
n *= 3;
priority_queue<int> q;
int x = 0;
for(int i = 0; i < n; ++i)
{
cin >> x;
q.push(x);
}
int res = 0;
for(int i = n / 3; i < n - n / 3; ++i)
{
q.pop();
res += q.top();
q.pop();
}
cout << res << endl;
return 0;
}
C++代码2
#include <iostream>
#include <algorithm>
#define int long long
using namespace std;
const int N = 1e5 + 10;
int arr[N * 3];
signed main()
{
int n = 0;
cin >> n;
for(int i = 0; i < 3 * n; i++)
{
cin >> arr[i];
}
sort(arr, arr + 3 * n);
int pos = 3 * n - 2, count = 1;
int ret = 0;
while(count++ <= n)
{
ret += arr[pos];
pos -= 2;
}
cout << ret << endl;
return 0;
}
Java代码
import java.util.*;
// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main
{
public static void main(String[] args)
{
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int[] arr = new int[n * 3];
for(int i = 0; i < n * 3; i++)
{
arr[i] = in.nextInt();
}
Arrays.sort(arr);
int pos = n * 3 - 2, count = 1;
long ret = 0;
while(count++ <= n)
{
ret += arr[pos];
pos -= 2;
}
System.out.println(ret);
}
}