华为od机试真题——智能成绩表

题目描述

算法结果

3 3
math chinese english
z3 71 81 91
l4 81 71 93
w5 21 91 95
math
l4 81 71 93 245 
z3 71 81 91 243 
w5 21 91 95 207 

算法详情

public class SmartScoreTable {
    public static void main(String[] args) {
        // 输入
        Scanner scanner = new Scanner(System.in);
        String[] inputNumber = scanner.nextLine().split(" ");
        // 学生人数
        int n = Integer.parseInt(inputNumber[0]);
        // 科目数量
        int m = Integer.parseInt(inputNumber[1]);
        // 科目
        String[] subjects = scanner.nextLine().split(" ");
        // 第一列为学生名字，最后一列为总分
        String[][] scores = new String[n][m + 2];
        for (int i = 0; i < n; i++) {
            String[] score = scanner.nextLine().split(" ");
            System.arraycopy(score, 0, scores[i], 0, score.length);
        }
        // 排序科目
        String sortSubject = scanner.nextLine();
        scanner.close();

        // 计算总分
        calTotalScore(scores);
        // 排序科目在科目数组中的下标
        int index = Arrays.stream(subjects).collect(Collectors.toList()).indexOf(sortSubject);
        List<String[]> result = Arrays.stream(scores).collect(Collectors.toList());
        result.sort(new Comparator<String[]>() {
            @Override
            public int compare(String[] o1, String[] o2) {
                return compareRule(o1, o2, index);
            }
        });

        // 输出
        for (String[] score : result) {
            String output = "";
            for (String str : score) {
                output = output + str + " ";
            }
            System.out.println(output);
        }

    }

    /**
     * 排序规则
     *
     * @param str1  分数字符数组1
     * @param str2  分数字符数组2
     * @param index 排序科目在科目数组中的下标
     * @return 比较结果
     */
    private static int compareRule(String[] str1, String[] str2, int index) {
        int len = str1.length;
        // index为-1,按总分排序
        if (index == -1) {
            // 总分相等，按名字排序
            if (Integer.parseInt(str2[len - 1]) - Integer.parseInt(str1[len - 1]) == 0) {
                return str1[0].compareTo(str2[0]);
            }
            return Integer.parseInt(str2[len - 1]) - Integer.parseInt(str1[len - 1]);
        } else {
            // index + 1为科目在分数数组中的下标，第一列为学生名
            if (Integer.parseInt(str2[index + 1]) - Integer.parseInt(str1[index + 1]) == 0) {
                return str1[0].compareTo(str2[0]);
            }
            return Integer.parseInt(str2[index + 1]) - Integer.parseInt(str1[index + 1]);
        }
    }

    /**
     * 计算学生总分
     *
     * @param scores 分数数组
     */
    private static void calTotalScore(String[][] scores) {
        int rows = scores.length;
        int cols = scores[0].length;
        for (int i = 0; i < rows; i++) {
            int total = 0;
            for (int j = 1; j < cols - 1; j++) {
                total += Integer.parseInt(scores[i][j]);
            }
            // 最后一列为总分
            scores[i][cols - 1] = Integer.toString(total);
        }
    }
}