我們有一個項的集合,其中第 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;}
}