LeetCode 1090. 受标签影响的最大值（优先队列）

示例 1：
输入：values = [5,4,3,2,1], labels = [1,1,2,2,3], 
num_wanted = 3, use_limit = 1
输出：9
解释：选出的子集是第一项，第三项和第五项。

示例 2：
输入：values = [5,4,3,2,1], labels = [1,3,3,3,2], 
num_wanted = 3, use_limit = 2
输出：12
解释：选出的子集是第一项，第二项和第三项。

示例 3：
输入：values = [9,8,8,7,6], labels = [0,0,0,1,1], 
num_wanted = 3, use_limit = 1
输出：16
解释：选出的子集是第一项和第四项。

示例 4：
输入：values = [9,8,8,7,6], labels = [0,0,0,1,1], 
num_wanted = 3, use_limit = 2
输出：24
解释：选出的子集是第一项，第二项和第四项。
 
提示：
1 <= values.length == labels.length <= 20000
0 <= values[i], labels[i] <= 20000
1 <= num_wanted, use_limit <= values.length

class Solution {
public:
    int largestValsFromLabels(vector<int>& values, vector<int>& labels, int num_wanted, int use_limit) {
    	vector<priority_queue<int>> countUse(20001);//
    	for(int i = 0; i < values.size(); ++i)
    		countUse[labels[i]].push(values[i]);
    	int k, sum = 0;
    	priority_queue<int> q;
    	for(int i = 0; i <= 20000; ++i)
    	{
    		k = use_limit;
    		while(k-- && !countUse[i].empty())
    		{
    			q.push(countUse[i].top());
    			countUse[i].pop();
    		}
    	}
    	while(num_wanted-- && !q.empty())
    	{
    		sum += q.top();
    		q.pop();
    	}
    	return sum;
    }
};