前言
Weekly Contest 141的 受标签影响的最大值:
我们有一个项的集合,其中第
i
项的值为values[i]
,标签为labels[i]
。我们从这些项中选出一个子集
S
,这样一来:
|S| <= num_wanted
- 对于任意的标签
L
,子集S
中标签为L
的项的数目总满足<= use_limit
。返回子集
S
的最大可能的 和。示例1:
输入:values = [5,4,3,2,1], labels = [1,1,2,2,3], num_wanted = 3, use_limit = 1 输出:9 解释:选出的子集是第一项,第三项和第五项。
示例2:
输入:values = [5,4,3,2,1], labels = [1,3,3,3,2], num_wanted = 3, use_limit = 2 输出:12 解释:选出的子集是第一项,第二项和第三项。
示例3:
输入:values = [9,8,8,7,6], labels = [0,0,0,1,1], num_wanted = 3, use_limit = 1 输出:16 解释:选出的子集是第一项和第四项。
示例4:
输入:values = [9,8,8,7,6], labels = [0,0,0,1,1], num_wanted = 3, use_limit = 2 输出:24 解释:选出的子集是第一项,第二项和第四项。
提示:
1 <= values.length == labels.length <= 20000
0 <= values[i], labels[i] <= 20000
1 <= num_wanted, use_limit <= values.length
解题思路
本题题目可能会有点难以理解,可用把题目中的项看做带有值的标签。而labels
就是标签列表,values
就是标签列表对应的标签值列表。
简单点说,标签与标签值的关系为多对多,一个标签可以有多个标签值,而一个相同的值可以多个标签都使用。
而题目的意思就是在所有项中选取num_wanted
个元素,但是选取的元素如果存在标签一致的情况,只能存在use_limit
个相同标签的项。找出满足上述条件的子集,且该子集的项的标签值之和最大,并返回该最大值。
下面我以示例4作为例子,讲解一下我的解题思路:
输入:values = [9,8,8,7,6], labels = [0,0,0,1,1], num_wanted = 3, use_limit = 2
输出:24
解释:选出的子集是第一项,第二项和第四项。
-
将
values
和labels
转化为以values
为key
,每个key
对应多个label
的结构,同时以key
从大到小进行排序。内容如下所示9 : [0] 8 : [0,0] 7 : [1] 6 : [1]
-
依次遍历第
1
步中的数据并进行处理,处理步骤如下:- 遍历
key
对应的label
列表 - 如果该
label
使用次数小于use_limit
,则将将最大值加上对应的key
,同时num_wanted
自减 - 每轮循环都要判断
num_wanted
是否为0
- 遍历
实现代码
/**
* 1090. 受标签影响的最大值
* @param values 标签值列表
* @param labels 标签名列表
* @param num_wanted 期待的标签个数
* @param use_limit 每个标签可重复的次数
* @return
*/
public int largestValsFromLabels(int[] values, int[] labels, int num_wanted, int use_limit) {
// value和label的映射,需要注意使用的是入参values作为key倒序(从大到小)存储
Map> valueLabelMap = new TreeMap<>(Comparator.reverseOrder());
// 每个label的出现次数
Map occurTimes = new HashMap<>();
// 最大值
int result = 0;
for(int i=0;i list=new ArrayList<>();
list.add(labels[i]);
valueLabelMap.put(values[i],list);
}
}
Iterator>> it=valueLabelMap.entrySet().iterator();
while (it.hasNext()){ // 遍历TreeMap
Map.Entry> entry = it.next();
// 对应的值
Integer value = entry.getKey();
List labelList = entry.getValue();
labelList.sort(Comparator.comparing(Integer::intValue));
for(int i=0;i