力扣：1282. 用户分组

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第15天，点击查看活动详情

1282. 用户分组

有 n 个人被分成数量未知的组。每个人都被标记为一个从 0 到 n - 1 的唯一ID 。

给定一个整数数组 groupSizes ，其中 groupSizes[i] 是第 i 个人所在的组的大小。例如，如果 groupSizes[1] = 3 ，则第 1 个人必须位于大小为 3 的组中。

返回一个组列表，使每个人 i 都在一个大小为 groupSizes[i] 的组中。

每个人应该 恰好只 出现在 一个组 中，并且每个人必须在一个组中。如果有多个答案，返回其中 任何 一个。可以 保证 给定输入 至少有一个 有效的解。

示例 1：
输入：groupSizes = [3,3,3,3,3,1,3]
输出：[[5],[0,1,2],[3,4,6]]
解释：
第一组是 [5]，大小为 1，groupSizes[5] = 1。
第二组是 [0,1,2]，大小为 3，groupSizes[0] = groupSizes[1] = groupSizes[2] = 3。
第三组是 [3,4,6]，大小为 3，groupSizes[3] = groupSizes[4] = groupSizes[6] = 3。 
其他可能的解决方案有 [[2,1,6],[5],[0,4,3]] 和 [[5],[0,6,2],[4,3,1]]。

示例 2：
输入：groupSizes = [2,1,3,3,3,2]
输出：[[1],[0,5],[2,3,4]]

提示：

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

问题解析

既然第i个人想留在总人数为groupSizes[i]的组里，那么也就是说，只有groupSizes[i]相同的人才能够放在一起。比如groupSizes[i]为2的人是不能和groupSizes[i]为3的人放在一起的，只能和其它groupSizes[i]为2的人分到一个组里面。

那么我们可以先用哈希表记录下来想留在人数为1~n的组里的人分别都有哪些。然后再把这些人分成总人数为groupSizes[i]的组。

比如groupSizes[i]为2的人有：1 2 3 4，那么我们可以把1 2分成一组，3 4分成一组。

最后再把所有的组存在一个vector里返回即可。

AC代码

class Solution {
public:
    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
        int n=groupSizes.size();
        map<int,vector<int>>mymap;
        for(int i=0;i<n;i++)
        {
            mymap[groupSizes[i]].push_back(i);
        }
        vector<vector<int>>res;
        for(auto i:mymap)
        {
            vector<int>ans;
            int len=i.second.size();
            for(int j=0;j<len;j++)
            {
                
                ans.push_back(i.second[j]);
                if(ans.size()==i.first)
                {
                    res.push_back(ans);
                    ans.clear();
                }
            }
        }
        return res;
    }
};