受标签影响的最大值

题目描述

我们有一个 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

对于useLimit，我们可以用hashmap将标签相同的元素存入一个list集合中，然后对集合进行排序，选取前useLimit个数加入最终进行选取的list中。
然后将最终的list进行排序选取前numWanted个数求和返回即可。

代码

class Solution {
    public int largestValsFromLabels(int[] values, int[] labels, int numWanted, int useLimit) {
        HashMap<Integer,List<Integer>> map=new HashMap<>();
        int n=values.length;
        for(int i=0;i<n;i++){
            if(map.containsKey(labels[i])==false){
                List<Integer> list=new ArrayList<>();
                list.add(values[i]);
                map.put(labels[i],list);
            }else{
                map.get(labels[i]).add(values[i]);
            }
        }
        List<Integer> li=new ArrayList<>();
        for(int k:map.keySet()){
            List<Integer> list1=map.get(k);
            Collections.sort(list1, new Comparator<Integer>() {
                @Override
                public int compare(Integer o1, Integer o2) {
                    return o2-o1;
                }
            });
            for(int j=0;j<list1.size() && j<useLimit;j++){
                li.add(list1.get(j));
            }
        }
        Collections.sort(li, new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });
        int res=0;
        for(int u=0;u<numWanted && u<li.size();u++){
            res+=li.get(u);
        }
        return res;
    }
}