(B卷,100分)- 统计射击比赛成绩（Java & JS & Python）

题目描述

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

条件如下

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

输入描述

• 输入第一行，一个整数N，表示该场比赛总共进行了N次射击，产生N个成绩分数（2<=N<=100）。
• 输入第二行，一个长度为N整数序列，表示参与每次射击的选手ID（0<=ID<=99）。
• 输入第三行，一个长度为N整数序列，表示参与每次射击的选手对应的成绩（0<=成绩<=100）。

输出描述

符合题设条件的降序排名后的选手ID序列。

用例

题目解析

简答的排序问题，按照题目要求写排序规则即可。

Java算法源码

import java.util.*;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);

    int n = Integer.parseInt(sc.nextLine());

    Integer[] ids =
        Arrays.stream(sc.nextLine().split(",")).map(Integer::parseInt).toArray(Integer[]::new);

    Integer[] scores =
        Arrays.stream(sc.nextLine().split(",")).map(Integer::parseInt).toArray(Integer[]::new);

    System.out.println(getResult(n, ids, scores));
  }

  public static String getResult(int n, Integer[] ids, Integer[] scores) {
    HashMap<Integer, ArrayList<Integer>> players = new HashMap<>();

    for (int i = 0; i < n; i++) {
      players.putIfAbsent(ids[i], new ArrayList<>());
      players.get(ids[i]).add(scores[i]);
    }

    ArrayList<int[]> ans = new ArrayList<>();

    for (int id : players.keySet()) {
      ArrayList<Integer> idScores = players.get(id);
      if (idScores.size() >= 3) {
        int total =
            idScores.stream().sorted((a, b) -> b - a).limit(3).reduce(Integer::sum).orElse(0);

        ans.add(new int[] {id, total});
      }
    }

    ans.sort((a, b) -> a[1] != b[1] ? b[1] - a[1] : b[0] - a[0]);

    StringJoiner sj = new StringJoiner(",");
    for (int[] player : ans) sj.add(player[0] + "");

    return sj.toString();
  }
}

JS算法源码

/* JavaScript Node ACM模式 控制台输入获取 */
const readline = require("readline");

const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout,
});

const lines = [];
rl.on("line", (line) => {
  lines.push(line);
  if (lines.length == 3) {
    const n = lines[0] - 0;
    const ids = lines[1].split(",").map(Number);
    const scores = lines[2].split(",").map(Number);

    console.log(getResult(n, ids, scores));

    lines.length = 0;
  }
});

function getResult(n, ids, scores) {
  const player = {};

  for (let i = 0; i < n; i++) {
    const id = ids[i];
    const score = scores[i];

    if (!player[id]) player[id] = [];
    player[id].push(score);
  }

  const ans = [];

  for (let id in player) {
    if (player[id].length >= 3) {
      const total = player[id]
        .sort((a, b) => b - a)
        .slice(0, 3)
        .reduce((a, b) => a + b);

      ans.push([id, total]);
    }
  }

  return ans
    .sort((a, b) => (a[1] != b[1] ? b[1] - a[1] : b[0] - a[0]))
    .map((x) => x[0])
    .join(",");
}

Python算法源码

# 输入获取
n = int(input())
ids = list(map(int, input().split(",")))
scores = list(map(int, input().split(",")))


# 算法入口
def getResult():
    players = {}

    for i in range(n):
        players.setdefault(ids[i], [])
        players.get(ids[i]).append(scores[i])

    ans = []

    for pid in players:
        if len(players[pid]) >= 3:
            players[pid].sort(reverse=True)
            ans.append((pid, sum(players[pid][:3])))

    ans.sort(key=lambda x: (-x[1], -x[0]))

    return ",".join(map(lambda x: str(x[0]), ans))


# 算法调用
print(getResult())