1090. Largest Values From Labels

class Solution {
    private static class Num implements Comparable<Num> {
        private final int value;
        private final int label;

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

        @Override
        public int compareTo(Num o) {
            return Integer.compare(o.value, this.value);
        }
    }

    public int largestValsFromLabels(int[] values, int[] labels, int numWanted, int useLimit) {
        Num[] nums = new Num[values.length];
        for (int i = 0; i < values.length; i++) {
            nums[i] = new Num(values[i], labels[i]);
        }
        Arrays.sort(nums);

        int numCount = 0;
        int sum = 0;
        Map<Integer, Integer> labelCountMap = new HashMap<>();
        for (Num num : nums) {
            if (numCount == numWanted) {
                break;
            }
            int labelCount = labelCountMap.getOrDefault(num.label, 0);
            if (labelCount == useLimit) {
                continue;
            }

            numCount++;
            sum += num.value;
            labelCountMap.put(num.label, labelCount + 1);
        }

        return sum;
    }
}

References

1090. Largest Values From Labels