题目描述：

1090. 受标签影响的最大值

提示

中等

46

相关企业

我们有一个 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

通过次数

9.6K

提交次数

15.1K

通过率

63.9%

题解分析：通过分析可以知道，这是一道贪心题目，于是可以归结到如何使用优先队列来解决。要求分数最高，那就是尽可能选择最大的value，如果没有其他限制条件，直接排序然后选择最大的numwanted个数就行。但是本题目中有限制，label不能相同，那么策略变为按照value排序，优先选择最大的，统计label个数，超过了限制，就选择次大的。于是，结合前面两篇博客，就写出来了这样的代码。

class Solution {
    typedef pair<int,int> PSI;
    struct cmp
    {
        bool operator()(PSI a, PSI b)
        {
            return a.first<b.first;
        }
    };
public:
    int largestValsFromLabels(vector<int>& values, vector<int>& labels, int numWanted, int useLimit) {
        int sum=0;
        int count=0;
        priority_queue<PSI,vector<PSI>,cmp> pq;
        
        int n=values.size();
        unordered_map<int,int> Hashmap;
        for(int i=0;i<n;i++)
        {
            pq.push({values[i],labels[i]});
        }
        while(!pq.empty())
        {
            int front = pq.top().first;   //key
            int end = pq.top().second;   //value
            pq.pop();
            Hashmap[end]++;
            if(Hashmap[end]<=useLimit)
            {
                sum+=front;
                count++;
            }
            if(count>=numWanted)
            {
                break;
            }
        }
        return sum;
    }
};

 代码中，构造一个最大堆，比较函数是value最大优先，统计label出现的次数使用hashmap，结束条件是选择的个数已经达到numwanted，然后跳出来。过程中使用sum保存最后的求和结果。

时间36 ms 击败33.72%  内存19.8 MB 击败 51.74%

如果题目的返回值变成一个数组，也是很容易的。