leetcode:1090. 受标签影响的最大值(贪心)

链接:https://leetcode-cn.com/problems/largest-values-from-labels/submissions/
贪心问题,label和value都按value从大到小排序,然后取满足条件的尽可能大的即可。
C++代码:

class Solution {
public:
    struct pair{
        int value;
        int label;
    };
    int largestValsFromLabels(vector<int>& values, vector<int>& labels, int num_wanted, int use_limit) {
        vector<pair> p;
        for(int i = 0;i<values.size();i++)
        {
            pair np{values[i],labels[i]};
            p.push_back(np);
        }
        sort(p.begin(), p.end(),
                [](pair i1, pair i2) {
                    return (i1.value>i2.value);
        });
        
        unordered_map<int, int> m;
        int sum = 0;
        int num_sum = 0;
        for(int i =0 ;i<values.size();i++)
        {
            if(num_sum>=num_wanted)
                return sum;
            else if(m[p[i].label] == use_limit)
                continue;
            else 
            {
                m[p[i].label]++;
                sum+=p[i].value;
                
                num_sum++;
            }
        }
        return sum;
    }
};

你可能感兴趣的:(leetcode,c++,leetcode,算法)