每日OJ题_牛客_组队竞赛_贪心_C++_Java

目录

牛客_组队竞赛_贪心

题目解析

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);
    }
}