我們有一個項的集合，其中第 i 項的值為 values[i]，標(biāo)簽為 labels[i]。

我們從這些項中選出一個子集 S，這樣一來：

|S| <= num_wanted
對于任意的標(biāo)簽 L，子集 S 中標(biāo)簽為 L 的項的數(shù)目總滿足 <= 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<>();//記錄相同標(biāo)簽出現(xiàn)的次數(shù)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;} }

總結(jié)

以上是生活随笔為你收集整理的leetcode1090. 受标签影响的最大值（贪心算法）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。