我们有一个项的集合,其中第 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;}
}