【Leetcode】1282. Group the People Given the Group Size They Belong To

题目地址：

https://leetcode.com/problems/group-the-people-given-the-group-size-they-belong-to/

给定一个长$n$的数组$A$，对于元素$0,1,...,n-1$这$n$个数存在一个划分，$A[i]$代表编号$i$所属的等价类的元素个数。要求返回$A$的划分。返回任意一个合法划分即可。

开个哈希表，key存等价类元素个数，value存所属等价类个数等于key的元素编号列表。容易证明value的长度是key的倍数。只需要将value的列表每key个划分为一组即可。代码如下：

class Solution {
 public:
  vector<vector<int>> groupThePeople(vector<int>& gs) {
    vector<vector<int>> res;
    unordered_map<int, vector<int>> mp;
    for (int i = 0; i < gs.size(); i++) mp[gs[i]].push_back(i);
    for (auto& [k, v] : mp)
      for (int i = 0; i < v.size() / k; i++)
        res.push_back(vector<int>(v.begin() + i * k, v.begin() + (i + 1) * k));
    return res;
  }
};

时空复杂度$O(n)$。