【Leetcode】1090. Largest Values From Labels（配数学证明）

题目地址：

https://leetcode.com/problems/largest-values-from-labels/description/

给定$n$个物品，每个物品有两个属性分别是其价值$v[i]$和其标签$l[i]$。价值和标签都是非负整数。给定两个正整数$m$和$u$，$m$表示总共最多能选择的物品数，$u$表示同一个标签能选择的最多物品数。问如何选择能使得总价值最大，返回最大总价值。

思路是贪心。直接按照价值从大到小对物品排序，然后遍历物品，依次看当前物品能不能加入，能加入就加。最后返回总价值即可。

思路证明比较简单。想象一下我们先将所有的物品按标签分组，每个标签内部再按价值从大到小排列，假设有$k$个分组，那么每个分组里只有价值最高的那$u$个可能被选中。我们想象每个分组中的价值就是数轴上的点，然后有$k$组点，这样我们用一个竖直的扫描线从数轴大的数向小的数的方向扫过去，如果扫到某个数的时候，扫描线穿过的所有分组的物品都能选，显然这些物品都加进去是最优解；如果所有分组的物品都选了会超$m$的个数，显然我们选哪个分组的物品是等效的。而上面的算法所做的就是这个步骤。

代码如下：

class Solution {
 public:
  using PII = pair<int, int>;
  int largestValsFromLabels(vector<int>& vs, vector<int>& ls, int numW,
                            int useL) {
    vector<int> idx;
    for (int i = 0; i < vs.size(); i++) idx.push_back(i);
	sort(idx.begin(), idx.end(), [&](int i, int j) {
      return vs[i] > vs[j];
    });

    unordered_map<int, int> mp;
    int res = 0, cnt = 0;
    for (int id : idx) {
      int l = ls[id], v = vs[id];
      if (mp[l] + 1 <= useL && cnt + 1 <= numW) {
        res += v;
        if (cnt + 1 == numW) break;
        cnt++;
        mp[l]++;
      }
    }

    return res;
  }
};

时间复杂度$O(n\log n)$，空间$O(n)$。