【CODE】Group the People Given the Group Size They Belong To

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

Medium

9546Add to ListShare

There are n people whose IDs go from 0 to n - 1 and each person belongs exactly to one group. Given the array groupSizes of length n telling the group size each person belongs to, return the groups there are and the people's IDs each group includes.

You can return any solution in any order and the same applies for IDs. Also, it is guaranteed that there exists at least one solution. 

Example 1:

Input: groupSizes = [3,3,3,3,3,1,3]
Output: [[5],[0,1,2],[3,4,6]]
Explanation: Other possible solutions are [[2,1,6],[5],[0,4,3]] and [[5],[0,6,2],[4,3,1]].

Example 2:

Input: groupSizes = [2,1,3,3,3,2]
Output: [[1],[0,5],[2,3,4]]

Constraints:

• groupSizes.length == n
• 1 <= n <= 500
• 1 <= groupSizes[i] <= n

class Solution {
public:
    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
        vector<vector<int> > res;
        unordered_map<int,vector<int> > umap;
        for(int i=0;i<groupSizes.size();i++){
            if(umap.find(groupSizes[i])==umap.end()){
                vector<int> tmp;
                umap[groupSizes[i]]=tmp;
            }
            umap[groupSizes[i]].push_back(i);
        }
        unordered_map<int,vector<int> >::iterator it;
        for(it=umap.begin();it!=umap.end();it++){
            int num=it->first;
            for(int i=0;i<it->second.size()/num;i++){
                vector<int> tmp;
                for(int j=i*num;j<(i+1)*num;j++){
                    tmp.push_back(it->second[j]);
                }
                res.push_back(tmp);
            }
        }
        return res;
    }
};

• Runtime: 28 ms, faster than 54.96% of C++ online submissions for Group the People Given the Group Size They Belong To.
• Memory Usage: 12.3 MB, less than 100.00% of C++ online submissions for Group the People Given the Group Size They Belong To.
• Next challenges:
• Create Maximum Number
• Set Intersection Size At Least Two
• String Without AAA or BBB