12-组队竞赛

题目

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


思路

由题意也就是让每3个一组的数字中的第2个数尽可能大,那么可以先对输入的所有数字按从小到达进行排序,然后在排序后的所有数字中:

  • 选取第1个数、倒数第1个数、倒数第2个数,组成一组;
  • 选取第2个数、倒数第3个数、倒数第4个数,组成一组;
  • 选取第3个数、倒数第5个数、倒数第6个数,组成一组;
  • ...

这样即可实现让每 3 个一组的数字中的第 2 个数尽可能大。

具体实现思路为:

在排序后的所有数字组成的数组arr中,从第一个数字开始遍历到第n个数字(n为队伍数),用i来表示当前遍历到的数字的索引(从0开始),那么在由索引为i的数字组成的3个数字中(索引为i的数字是这3个数字中最小的,也是第一个),中位数为:arr[arr.length-2*(i+1)]。

例1:

  • 现在排序后有 1 2 5 5 8 9
  • 那么分组为1 8 9 和 2 5 5

例2:

  • 现在排序后有 1 2 3 4 5 6 7 8 9
  • 那么分组为1 8 9 和 2 6 7 和 3 4 5

代码

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while(sc.hasNextInt()) {
            //输入队伍数n
            int n = sc.nextInt();

            //依次输入所有队伍的每个参赛选手的水平值,总共有3 * n个,用arr数组接收
            long[] arr = new long[3 * n];
            for(int i = 0; i < (3 * n); i++) {
                arr[i] = sc.nextLong();
            }

            //对所有队伍的参赛选手的水平值组成的数组进行升序排序
            Arrays.sort(arr);

            //定义变量sum,作为最后的返回结果,即所有队伍的水平值总和的最大值.
            long sum = 0;
            //遍历排序后的数组,相加累计求得所有队伍的水平值总和的最大值.
            for(int i = 0; i < n; i++) {
                sum += arr[arr.length - 2 * (i + 1)];
            }
            System.out.println(sum);
        }
    }
}

你可能感兴趣的:(笔试刷题-编程题,算法)