受标签影响的最大值

/*
 * 思路：
 * 1. 创建一个数组表示每个元素的值和标签。
 * 2. 按照值从大到小对元素进行排序。
 * 3. 使用一个哈希表记录每个标签的使用次数。
 * 4. 使用Stream API遍历排序后的数组，选择不超过 numWanted 个元素，并且每个标签不超过 useLimit 次。
 * 5. 返回选择的元素值之和。
 */
import java.util.*;
import java.util.stream.*;

public class Solution {
    public int largestValsFromLabels(int[] values, int[] labels, int numWanted, int useLimit) {
        int n = values.length;
        List<int[]> items = IntStream.range(0, n)
                                      .mapToObj(i -> new int[]{values[i], labels[i]})
                                      .sorted((a, b) -> Integer.compare(b[0], a[0]))
                                      .collect(Collectors.toList());
        Map<Integer, Long> labelCount = new HashMap<>();
        int[] sum = {0};
        int[] count = {0};
        items.stream()
             .filter(item -> {
                 int label = item[1];
                 labelCount.put(label, labelCount.getOrDefault(label, 0L) + 1);
                 if (labelCount.get(label) <= useLimit && count[0] < numWanted) {
                     sum[0] += item[0];
                     count[0]++;
                     return true;
                 } else {
                     labelCount.put(label, labelCount.get(label) - 1);
                     return false;
                 }
             })
             .collect(Collectors.toList());
        return sum[0];
    }
}