Largest Values From Labels

You are given n item's value and label as two integer arrays values and labels. You are also given two integers numWanted and useLimit.

Your task is to find a subset of items with the maximum sum of their values such that:

  • The number of items is at most numWanted.
  • The number of items with the same label is at most useLimit.

Return the maximum sum.

Example 1
Inputvalues = [5,4,3,2,1], labels = [1,1,2,2,3], numWanted = 3, useLimit = 1
Output9
The subset chosen is the first, third, and fifth items with the sum of values 5 + 3 + 1.
Example 2
Inputvalues = [5,4,3,2,1], labels = [1,3,3,3,2], numWanted = 3, useLimit = 2
Output12
The subset chosen is the first, second, and third items with the sum of values 5 + 4 + 3.

Constraints

  • n == values.length == labels.length
  • 1 <= n <= 2 * 10^4
  • 0 <= values[i], labels[i] <= 2 * 10^4
  • 1 <= numWanted, useLimit <= n

Asked at 1 companies

</>

Your Solution

(Ctrl/Cmd + Enter)

Switching Language

Loading template...

Loading...

Sign in to save your progress

AI code evaluation

Get a correctness verdict, missed edge cases, and complexity analysis of your solution.

Sign in to evaluate