前五科的均分

/*
 * 题解思路:
 * 1. 使用Java Stream API来处理数据。
 * 2. 将输入数据转换为Map，其中键是学生ID，值是分数列表。
 * 3. 对分数列表进行排序并取前五个分数。
 * 4. 计算平均分并返回结果。
 */
import java.util.*;
import java.util.stream.*;
public class Solution {
    public int[][] highFive(int[][] items) {
        return Arrays.stream(items)
                .collect(Collectors.groupingBy(
                        item -> item[0],
                        Collectors.mapping(item -> item[1], Collectors.toList())
                ))
                .entrySet().stream()
                .map(entry -> {
                    List<Integer> scores = entry.getValue().stream()
                            .sorted(Collections.reverseOrder())
                            .limit(5)
                            .collect(Collectors.toList());
                    int avg = (int) scores.stream().mapToInt(Integer::intValue).average().orElse(0);
                    return new int[]{entry.getKey(), avg};
                })
                .toArray(int[][]::new);
    }
}