【牛客刷题系列】贪心算法相关面试题总结

组队竞赛

链接：组队竞赛
来源：牛客网

牛牛举办了一次编程比赛,参加比赛的有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(1 ≤ n ≤ 10^5)

第二行包括3*n个整数a_i(1 ≤ a_i ≤ 10^9),表示每个参赛选手的水平值.

输出描述:
输出一个整数表示所有队伍的水平值总和最大值.

示例1

输入
2
5 2 8 5 1 5
输出
10

题目解析：

队伍的水平值等于该队伍队员中第二高水平值，为了所有队伍的水平值总和最大的解法，也就是说每个队伍的第二个值是尽可能大的值。所以实际值把最大值放到最右边，最小是放到最左边，就是先进行排序。

本题的主要思路是贪心算法，贪心算法其实很简单，就是每次选值时都选当前能看到的局部最解忧，所以这里的贪心就是保证每组的第二个值取到能选择的最大值就可以，我们每次尽量取最大，但是要满足题目条件，这里要的是第二个大的数，所以对所有数先进行排序，再从右到左依次拿到3n-2、3n-4…直到n.
比如示例中的数字：5 2 8 5 1 5，进行排序后，得到 1 2 5 5 5 8，因此拿到5 + 5 = 10.

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

int main()
{
    int n=0;
    cin>>n;
    vector<int> v(3*n);
    for(int j=0;j<3*n;j++)
    {
        cin>>v[j];
    }
    sort(v.begin(),v.end());
    long long  num=0;
    int beg=3*n-2;
    while(n--)
    {
        num+=v[beg];
        beg-=2;
    }
    cout<<num<<endl;
    return 0;
}