受标签影响的最大值

题目：

我们有一个 n 项的集合。给出两个整数数组 values 和 labels ，第 i 个元素的值和标签分别是 values[i] 和 labels[i]。还会给出两个整数 numWanted 和 useLimit 。

从 n 个元素中选择一个子集 s :

• 子集 s 的大小 小于或等于 numWanted 。
• s 中 最多 有相同标签的 useLimit 项。
• 一个子集的 分数 是该子集的值之和。

返回子集 s 的最大 分数 。

示例:

输入：values = [5,4,3,2,1], labels = [1,1,2,2,3], numWanted = 3, useLimit = 1
输出：9
解释：选出的子集是第一项，第三项和第五项。

代码：

package com.lixusheng.interview;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

public class Main {
    static class Item {

        public Item(int value, int label) {
            this.value = value;
            this.label = label;
        }
        int value;
        int label;

    }

    public static int largestValsFromLabels(int[] values, int[] labels, int numWanted, int useLimit) {

        List<Item> items = new ArrayList<>();
        Item item;
        for (int i = 0; i < values.length; i++) {
            item = new Item(values[i], labels[i]);
            items.add(item);
        }

        items.sort((o1, o2) -> o2.value - o1.value)10;

        //计算列表中重复的个数
        int maxScore = 0;
        int[] repeatCount = new int[20001];
        int count = 0;
        for (int i = 0; i < items.size(); i++) {
            Item tempItem = items.get(i);

            if (++repeatCount[tempItem.label] > useLimit) {
                continue;
            }
            if (count + 1 > numWanted) {
                break;
            }
            count++;
            maxScore = maxScore + tempItem.value;
        }
        return maxScore;
    }

    public static void main(String[] args) {
        int[] values = {5, 4, 3, 2, 1};
        int[] labels = {1, 1, 2, 2, 3};
        int numWanted = 3, useLimit = 1;
        System.out.println(largestValsFromLabels(values, labels, numWanted, useLimit));
    }

}