1282. Group the People Given the Group Size They Belong To

题目传送门

问题分析

按照问题描述，我们可以得出如下结论。假设一个人声明自己所属的组的数目为3，那么我们将groupSize中的3全部放在一起，其数目一定是3的倍数，并且我们只要从这些数字中依次取3个便可以构成一个合法的组。
其中算法中利用了hashmap结构，用于映射元素，保证在线性时间内找到相应的元素。

代码

class Solution {
public:
    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
        unordered_map<int, vector<int>> groupMap;
        vector<vector<int>> result;
        for(int i = 0; i < groupSizes.size(); i++){
            groupMap[groupSizes[i]].push_back(i);
            if(groupMap[groupSizes[i]].size() == groupSizes[i]){
                result.push_back(groupMap[groupSizes[i]]);
                groupMap[groupSizes[i]] = {};
            }
        }
        return result;
    }
};