LC-1090. 受标签影响的最大值（计数贪心）

1090. 受标签影响的最大值

难度中等34

我们有一个 n 项的集合。给出两个整数数组 values 和 labels ，第 i 个元素的值和标签分别是 values[i] 和 labels[i]。还会给出两个整数 numWanted 和 useLimit 。

从 n 个元素中选择一个子集 s :

• 子集 s 的大小 小于或等于 numWanted 。
• s 中 最多 有相同标签的 useLimit 项。

一个子集的 分数 是该子集的值之和。

返回子集 s 的最大 分数 。

示例 1：

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

示例 2：

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

示例 3：

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

提示：

• n == values.length == labels.length
• 1 <= n <= 2 * 104
• 0 <= values[i], labels[i] <= 2 * 104
• 1 <= numWanted, useLimit <= n

计数贪心

按从大到小的顺序遍历labels，如果当前labels[idx[i]]计数次数超过了uselimit，则说明这个数选到上限了，continue，否则加上这个数然后计数 + 1

class Solution {
    public int largestValsFromLabels(int[] values, int[] labels, int numWanted, int useLimit) {
        int cnt[] = new int[20005];
        int n = labels.length;
        Integer idx[] = new Integer[n];
        for(int i = 0; i < n; i++)
            idx[i] = i;
        Arrays.sort(idx, (a, b) -> values[b] - values[a]);
        int ans = 0;
        for(int i = 0; i < n && numWanted > 0; i++, numWanted--){
            while(i < n && cnt[labels[idx[i]]] >= useLimit)
                i++;
            if(i < n){
                ans += values[idx[i]];
                cnt[labels[idx[i]]] += 1;
            }
        }
        return ans;
    }
}