牛客——组队竞赛

链接： https://www.nowcoder.com/questionTerminal/248ccf8b479c49a98790db17251e39bb
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 32M，其他语言64M

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

解题思路：

本题需要先将选手的水平值进行排序，比如排序好的所有选手水平值为1、2、3……n、n+1、n+2、n+3

因为一个队伍的水平值是队员中的第二高水平值，所以不能取最大值，那就可以在第二大的开始取，n+2、n、n-2……这样就在后2n个数中取出了n个第二大的值，将这n个值相加就是要求的最大水平值总和

排序我相信很多人都能想到，于是我就写出了如下代码

#include <stdio.h>
#include<stdlib.h>

void Swap(int* x, int* y)
{
    int tmp = *x;
    *x = *y;
    *y = tmp;
}

void sort(int* a, int size)
{
    int index=0;
    for(int i=0;i<size;i++)
    {
        index=i;
        for(int j=i+1;j<size;j++)
        {
            if(a[index] > a[j])
                index=j;
        }
        if(i!=index)
        {
            Swap(&a[i], &a[index]);
        }
    }
}

int main() {
    int n;
    scanf("%d", &n);
    int* arr = (int*)malloc(sizeof(int) * 3 * n);
    for (int i = 0; i < 3 * n; i++) {
        scanf("%d", &arr[i]);
    }
    sort(a,3*n);
    long long sum = 0;
    for (int i = n; i < 3 * n; i += 2) {
        sum += arr[i];
    }
    printf("%lld", sum);
    return 0;
}

选择用简单的冒泡或选择的方式进行排序，这样运行之后会报这样的错误

会提示运行超时，这是因为使用的排序方式时间复杂度过大，导致超出时间限制，所以可以使用效率更高的排序方式

如果将上面代码中的sort函数换成快速排序的quicksort函数，就可以运行成功

void quicksort(int* a, int begin, int end)
{
    if (begin >= end)
        return;

    int cur = begin, pre = begin + 1;
    
    int key=a[begin];

    while (pre <= end)
    {
        while (a[pre] <= key && ++cur != pre)
            Swap(&a[cur], &a[pre]);
        pre++;
    }
    Swap(&a[begin], &a[cur]);

    quicksort(a, begin, cur - 1);
    quicksort(a, cur + 1, end);
}

//上面代码的第36行改成这个-> quicksort(arr, 0, 3 * n - 1);

注意点：

1. 排序方式的选择——时间复杂度大会超出时间限制

2. sum的类型要设置成long long类型——设置成int，数据大且多，会超出整形范围