LeetCode 1086. 前五科的均分（map + 优先队列）

1. 题目

给你一个不同学生的分数列表，请按 学生的 id 顺序 返回每个学生 最高的五科 成绩的 平均分。

对于每条 items[i] 记录， items[i][0] 为学生的 id，items[i][1] 为学生的分数。
平均分请采用整数除法计算。

示例：
输入：[[1,91],[1,92],[2,93],[2,97],[1,60],[2,77],[1,65],[1,87],[1,100],[2,100],[2,76]]
输出：[[1,87],[2,88]]
解释：
id = 1 的学生平均分为 87。
id = 2 的学生平均分为 88.6。但由于整数除法的缘故，平均分会被转换为 88。
 
提示：
1 <= items.length <= 1000
items[i].length == 2
学生的 ID 在 1 到 1000 之间
学生的分数在 1 到 100 之间
每个学生至少有五个分数

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/high-five
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• 每个人对应的 优先队列存入最大的5个成绩

class Solution {
public:
    vector<vector<int>> highFive(vector<vector<int>>& items) {
    	vector<vector<int>> ans;
    	map<int,priority_queue<int,vector<int>,greater<int>>> m;
    	int id, v;
    	for(auto& i : items)
    	{
    		id = i[0];
    		v = i[1];
    		if(m[id].size()<5)
    			m[id].push(v);
    		else if(m[id].size()==5 && m[id].top() < v)
    		{
    			m[id].pop();
    			m[id].push(v);
    		}
    	}
    	for(auto it = m.begin(); it != m.end(); ++it)
    	{
    		id = it->first;
    		v = 0;
    		while(!m[id].empty())
    		{
    			v += m[id].top();
    			m[id].pop();
    		}
    		ans.push_back({id,v/5});
    	}
    	return ans;
    }
};

28 ms 9.5 MB

---

长按或扫码关注我的公众号，一起加油、一起学习进步！