【Leetcode】1086. High Five

题目地址：

https://leetcode.com/problems/high-five/

给定一个二维数组，每一行代表某个学生的某一科目分数，第一个数字是学生id，第二个数字是分数。要求返回一个数组，存储每个学生最高的$5$门课的平均分（返回整型即可），并且数组要按学生id排序。

思路是哈希表 + 堆。哈希表key存id，value存一个堆，堆里存id对应的学生的最高无门课的分数。代码如下：

import java.util.Map;
import java.util.PriorityQueue;
import java.util.TreeMap;

public class Solution {
    public int[][] highFive(int[][] items) {
    	// 用TreeMap就可以在遍历的时候按照id从小到大遍历了
        Map<Integer, PriorityQueue<Integer>> map = new TreeMap<>();
        for (int[] item : items) {
            map.putIfAbsent(item[0], new PriorityQueue<>());
            PriorityQueue<Integer> minHeap = map.get(item[0]);
            if (minHeap.size() < 5) {
                minHeap.offer(item[1]);
            } else {
                if (item[1] > minHeap.peek()) {
                    minHeap.poll();
                    minHeap.offer(item[1]);
                }
            }
        }
     
        int[][] res = new int[map.size()][2];
        int idx = 0;
        for (Map.Entry<Integer, PriorityQueue<Integer>> entry : map.entrySet()) {
            res[idx][0] = entry.getKey();
            int sum = 0;
            for (int score : entry.getValue()) {
                sum += score;
            }
            
            res[idx][1] = sum / 5;
            idx++;
        }
        return res;
    }
}

时间复杂度$O(n)$，空间$O(n_{id})$，$n$为数组长度，$n_{id}$为学生数量。