leetcode1090. 受标签影响的最大值（贪心算法）

我们有一个项的集合，其中第 i 项的值为 values[i]，标签为 labels[i]。

我们从这些项中选出一个子集 S，这样一来：

|S| <= num_wanted
对于任意的标签 L，子集 S 中标签为 L 的项的数目总满足 <= use_limit。
返回子集 S 的最大可能的 和。

示例 1：

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

代码

class Solution {
    public int largestValsFromLabels(int[] values, int[] labels, int num_wanted, int use_limit) {

        PriorityQueue<int[]> priorityQueue=new PriorityQueue<>(((o1, o2) -> o2[0]-o1[0]));
//堆排序
        for(int i=0;i<values.length;i++)
        {
            priorityQueue.offer(new int[]{values[i],labels[i]});
        }
        Map<Integer,Integer> map=new HashMap<>();//记录相同标签出现的次数
        int res=0,num=0;
        while (!priorityQueue.isEmpty())//贪心算法
        {
            int[] temp=priorityQueue.poll();
            if(map.containsKey(temp[1])&&map.get(temp[1])+1>use_limit) continue;
            map.put(temp[1],map.getOrDefault(temp[1],0)+1);
            res+=temp[0];
               num++;
              if(num==num_wanted) break;
        }
        return res;
    }
}