【算法题】统计射击比赛成绩

351次阅读

共计 1853 个字符，预计需要花费 5 分钟才能阅读完成。

给定一个射击比赛成绩单，包含多个选手若干次射击的成绩分数，请对每个选手按其最高三个分数之和进行降序排名，输出降序排名后的选手ID序列。

一个选手可以有多个射击成绩的分数，且次序不固定
如果一个选手成绩少于3个，则认为选手的所有成绩无效，排名忽略该选手
如果选手的成绩之和相等，则相等的选手按照其ID降序排列

题目要求
输入第一行，一个整数N，表示该场比赛总共进行了N次射击，产生N个成绩分数（2<=N<=100）
输入第二行，一个长度为N整数序列，表示参与每次射击的选手id（0<=id<=99）
输入第三行，一个长度为N整数序列，表示参与每次射击选手对应的成绩（0<=成绩<=100）
输出符合题设条件的降序排名后的选手ID序列

示例1

输入：
13
3,3,7,4,4,4,4,7,7,3,5,5,5
53,80,68,24,39,76,66,16,100,55,53,80,55
输出：5,3,7,4
说明:
该场射击比赛进行了13次，参赛的选手为{3,4,5,7}
3号选手成绩53,80,55 最高三个成绩的和为188
4号选手成绩24,39,76,66 最高三个成绩的和为205
5号选手成绩53,80,55 最高三个成绩的和为188
7号选手成绩68,16,100 最高三个成绩的和为184
比较各个选手最高3个成绩的和，有4号>3号=5号>7号，由于3号和5号成绩相等，且id 5>3，所以输出 7,5,3,4

import java.util.*;
import java.util.stream.Collectors;

/**
 * @since 2022年4月18日
 */
public class ShootCompetition {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        // 总射击次数
        int count = scanner.nextInt();
        // 射击选手的ID
        List<Integer> ids = Arrays.stream(scanner.nextLine().split(",")).map(Integer::parseInt).collect(Collectors.toList());
        // 射击选手对应的成绩
        List<Integer> scores = Arrays.stream(scanner.nextLine().split(",")).map(Integer::parseInt).collect(Collectors.toList());

        HashMap<Integer, List<Integer>> id_scores = new HashMap<>();
        // 分配选手的各自的分数
        for (int i = 0; i < count; i++) {
            Integer id = ids.get(i);
            Integer score = scores.get(i);
            List<Integer> list = id_scores.getOrDefault(id, new LinkedList<>());
            list.add(score);
            id_scores.put(id, list);
        }

        StringBuilder builder = new StringBuilder();
        id_scores.entrySet()
                .stream()
                .filter(x -> x.getValue().size() >= 3)
                .sorted((o1, o2) -> {
                    Integer sum1 = sum(o1.getValue());
                    Integer sum2 = sum(o2.getValue());
                    if (Objects.equals(sum1, sum2)) {
                        return o2.getKey() - o1.getKey();
                    } else {
                        return sum2 - sum1;
                    }
                })
                .map(Map.Entry::getKey)
                .forEach(x -> builder.append(x).append(","));

        System.out.println(builder.substring(0, builder.length() - 1));

    }

    private static Integer sum(List<Integer> list) {
        list.sort(Integer::compareTo);
        int sum = 0;
        for (int i = list.size() - 1; i >= list.size() - 3; i--) {
            sum += list.get(i);
        }
        return sum;
    }
}

正文完