目录
牛客_组队竞赛_贪心
题目解析
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);}
}