LeetCode每日一题--受标签影响的最大值

LeetCode每日一题--受标签影响的最大值_第1张图片
题目如图所示
题目大意:

一共可以选中numWanted个标签并使得他们的 价值(value) 最大,但是每个labels最多选中useLimit
实例1中只能选择同一个标签值一次,因此选择labels = 1 2 3的价值为5 3 1的三个元素得到最大价值9

  1. 因此我们可以使用一个二元容器存储数据
    vector> mp
    对其进行排序,要按照价值从大到小排序,我们可以将价值置为负数 这样就能够将其按照从大到小的价值进行排序
  2. 存储数据进行排序后,使用哈希表记录该标签值的使用次数。
  3. 得到最大结果

放代码!!!

typedef pair<int,int> PII;
class Solution {
public:
    int largestValsFromLabels(vector<int>& values, vector<int>& labels, int numWanted, int useLimit) {
        int n = values.size();
        vector<pair<int,int>> mp(n + 1);
        int res = 0;
        for(int i = 0; i < n; ++i) mp.push_back({-values[i], labels[i]});
        sort(mp.begin(), mp.end());
        unordered_map<int,int> M;
        int k = 0;
        for(int i = 0; i < n && k < numWanted; ++i) { 
            auto &[x, y] = mp[i];
            if(M[y] < useLimit) { // 是否满足条件
                res += -x; // 存入的价值是负数
                k++;       // 总标签数++
                M[y]++;    // 使用次数++
            }
        }
        return res;
    }
};

感谢阅读

你可能感兴趣的:(leetcode刷题笔记,leetcode,算法,哈希算法,排序算法)