持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第27天，点击查看活动详情

一、题目描述：

我们有一个 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 * 10^4
0 <= values[i], labels[i] <= 2 * 10^4
1 <= numWanted, useLimit <= n

二、思路分析：

把两个List合并成一个，每个元素都是一个[value,label] 然后将新List按照value排序

接下来就是循环，从做向右取满足条件的值，因为已经排序过，所以结果必然是最大的条件有两个：一个是num_wanted，计数器一个是use_limit，这个其实是标签的使用次数，用一个字典记录使用标签的次数

三、AC 代码：

class Solution:
    def largestValsFromLabels(self, values: List[int], labels: List[int], num_wanted: int, use_limit: int) -> int:
        together = [[values[i],labels[i]] for i in range(len(values))]
        together.sort(key= lambda x: x[0], reverse=True)
        res = 0
        i = 0
        label_dict = {}
        while i < len(together) and num_wanted:
            if together[i][1] not in label_dict:
                label_dict[together[i][1]] = 1
                num_wanted -= 1
                res += together[i][0]
            else:
                if label_dict[together[i][1]] < use_limit:
                    label_dict[together[i][1]] += 1
                    num_wanted -= 1
                    res += together[i][0]
            i += 1
        return res

四、参考：

用pair保存value和lable，然后对value进行排序。用hash表记录每个lable出现的次数。 - 受标签影响的最大值 - 力扣（LeetCode）

LeetCode 1090. 受标签影响的最大值

一、题目描述：

二、思路分析：

三、AC 代码：

四、参考：