题目如图所示
题目大意:
一共可以选中
numWanted
个标签并使得他们的 价值(value) 最大,但是每个labels
最多选中useLimit
次
实例1中只能选择同一个标签值一次,因此选择labels = 1 2 3
的价值为5 3 1
的三个元素得到最大价值9
- 因此我们可以使用一个二元容器存储数据
vector<pair<int,int>> mp
对其进行排序,要按照价值从大到小排序,我们可以将价值置为负数 这样就能够将其按照从大到小的价值进行排序 - 存储数据进行排序后,使用哈希表记录该标签值的使用次数。
- 得到最大结果
放代码!!!
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;
}
};
感谢阅读