我们有一个 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
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/largest-values-from-labels
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题过程
首先,要先看懂题目,我刚开始看的也不是很懂
有value值数组 label标签数组
则在numwanted个数之内,和uselimit最大重复个数的限制之下
找到子集s和的最大值
看看例子
values = [5,4,3,2,1], labels = [1,1,2,2,3], numWanted = 3, useLimit = 1
首先要找最大值,可以直接理解要找numwanted个数加起来最大,那很容易联想到是要先对values进行排序对吧,刚好这个就是逆序排列的
然后选出3个,理想情况就是取最前面三个5,4,3,但是还有个标签的限制,规定了uselimit是1,所以5,4只能2选1,因为标签1的只能由一个,那必然是5,所以可以的出本题的s所取得是
5,3,1=9
代码思路
1.要对value进行排序,但是value和label是一一对应得,所以可以联想到新建一个数组,记录value排序后的下标,则利用这个下标就可以取到,value和label所对应的值,
let index=[]
for(let i in values){
index.push(i)
}
index.sort((a,b)=>{
return values[b]-values[a]
})
2.采用Hash键值对的结构,来判断这个标签是否用过,健对应标签,值对应是否存在于子集中,
let map=new Map()
let sum=0
let i=0
while(i<values.length&&numWanted>0){
let idx=index[i]//因为index是下标数组,第一位显示的是value数组中最大的哪一位的下标。
if(map.has(labels[idx])){//判断如果在hash中存在的话
if(map.get(labels[idx])<useLimit){//获取到对应的值,看是否满足题目中对标签的约束
sum+=values[idx]//放进子集中
map.set(labels[idx],map.get(labels[idx])+1)//更新hash的大小
numWanted--//个数-1
}
}else{
map.set(labels[idx],1)//如果标签不存在,说明是第一次用在子集中。则直接赋值1
numWanted--//个数-1
sum+=values[idx]//放进子集中
}
i++
}