我们有一个 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
public int largestValsFromLabels(int[] values, int[] labels, int numWanted, int useLimit) {
int n = values.length,res=0;
Integer[] index = new Integer[n];
for (int i = 0; i < n; i++) {
index[i]=i;
}
Arrays.sort(index,(a,b)->values[b]-values[a]);
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < n&&numWanted>0; i++,numWanted--) {
while (i<n&&map.getOrDefault(labels[index[i]],0)>=useLimit){
i++;
}
if (i<n){
res+=values[index[i]];
map.put(labels[index[i]],map.getOrDefault(labels[index[i]],0)+1);
}
}
return res;
}
func largestValsFromLabels(values []int, labels []int, numWanted int, useLimit int) int {
n,res := len(values),0
//排序失败
//sort.Slice(index, func(i, j int) bool {
// return values[i]>values[j]
//})
for i := 0; i < n; i ++ {
minIdx := i
for j := i+1; j < n; j ++ {
if values[j] > values[minIdx] { //找最小的数
minIdx = j //保存最小数索引
}
}
values[i], values[minIdx] = values[minIdx],values[i]
labels[i], labels[minIdx] = labels[minIdx],labels[i]
}
mapX:=make(map[int]int,0)
for i := 0; i < n&&numWanted>0;i++ {
for i<n&&mapX[labels[i]]>=useLimit {
i++
}
if i<n{
res+=values[i]
mapX[labels[i]]++
}
numWanted--
}
return res
}