笔试 - 组队竞赛

组队竞赛
牛牛举办了一次编程比赛,参加比赛的有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),表示每个参赛选手的水平值.
输出描述:
输出一个整数表示所有队伍的水平值总和最大值.

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

思路：先排序，假设三组为 1 2 3 4 5 6 7 8 9则
1 8 9
2 6 7
3 4 5
队伍的水平值等于该队伍队员中第二高水平值

10^5 * 10^9 用int 溢出，long ans = 0;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] a = new int[n*3];
        for (int i = 0; i < n*3; i++)
            a[i] = sc.nextInt();
        Arrays.sort(a);
        long ans = 0;
        for (int i = n; i <= n * 3 - 2; i+=2) 
            ans += a[i];
        System.out.println(ans);
    }
}

手撕快排，练练。

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] a = new int[n*3];
        for (int i = 0; i < n*3; i++)
            a[i] = sc.nextInt();
        Main main = new Main();
        main.quickSort(a, 0, a.length - 1);
        long ans = 0;
        for (int i = n; i <= n * 3 - 2; i+=2) {
            ans += a[i];
        }
    }

    public void  quickSort(int[] a, int low, int high) {
        if (low < high){
            int mid = partition(a, low, high);
            quickSort(a, low, mid - 1);
            quickSort(a,mid + 1, high);
        }
    }

    public int partition(int[] a, int low, int high) {
        int base = a[low];
        int i = low;
        int j = high;
        while (i < j){
            while (i < j && a[j] > base)
                j--;
            while (i < j && a[i] <= base)
                i++;
            if (i < j){
                int t = a[i];
                a[i] = a[j];
                a[j] = t;
                i++;
                j--;
            }
        }
        if (a[i] > base){
            int t = a[i-1];
            a[i-1] = a[low];
            a[low] = t;
            return i-1;
        }
        int t = a[i];
        a[i] = a[low];
        a[low] = t;
        return i;
    }
}

2020/02/27 牛客网模拟笔试编程题的第二题
当时没看这题，太“蒟蒻”了，这题原来这么简单。