【LetMeFly】1090.受标签影响的最大值

力扣题目链接：https://leetcode.cn/problems/largest-values-from-labels/

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

方法一：贪心 + 排序

首先咱们不考虑这道题的“标签”，给你一个values数组和一个整数numWanted，让你从values中选取不超过numWanted个元素使得所选元素的和尽可能地大，那么应该怎么做呢？

很简单，将values中的元素按从大到小的规则排序，从前到后选择min(所有, numWanted)个元素即可。

现在仅仅是在刚才的基础上，添加了“标签”，并且做了以下限制：同一个标签的元素选取数量不超过useLimit个。

怎么办呢？很简单，我们仍然按照values数组从大到小的规则排序，在选取的过程中使用哈希表统计每个标签选择了多少个，如果某个标签已经选择了useLimit个，跳过这个元素继续选下一个元素就好了。

• 时间复杂度$O(n\log n)$，其中$n=len(values)$
• 空间复杂度$O(n)$

AC代码

C++

class Solution {
    
    
public:
    int largestValsFromLabels(vector<int>& values, vector<int>& labels, int numWanted, int useLimit) {
    
    
        vector<pair<int, int>> v(values.size());
        for (int i = 0; i < values.size(); i++) {
    
    
            v[i] = {
    
    values[i], labels[i]};
        }
        sort(v.begin(), v.end(), [&](const pair<int, int>& a, const pair<int, int>& b) {
    
    
            return a.first > b.first;
        });
        unordered_map<int, int> ma;
        int cnt = 0;
        int ans = 0;
        for (auto&& [value, label] : v) {
    
    
            if (ma[label] == useLimit) {
    
    
                continue;
            }
            ma[label]++;
            cnt++;
            ans += value;
            if (cnt == numWanted) {
    
    
                break;
            }
        }
        return ans;
    }
};

Python

# from typing import List
# from collections import defaultdict


class Solution:
    def largestValsFromLabels(self, values: List[int], labels: List[int], numWanted: int, useLimit: int) -> int:
        v = list(zip(values, labels))
        v.sort(key=lambda x : -x[0])
        ans = 0
        cnt = 0
        ma = defaultdict(int)
        for value, label in v:
            if ma[label] == useLimit:
                continue
            ma[label] += 1
            cnt += 1
            ans += value
            if cnt == numWanted:
                break
        return ans

同步发文于CSDN，原创不易，转载请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/130824464