LeetCode/用户分组

有 n 个人被分成数量未知的组。每个人都被标记为一个从 0 到 n - 1 的唯一ID 。
给定一个整数数组 groupSizes ，其中 groupSizes[i] 是第 i 个人所在的组的大小
例如，如果 groupSizes[1] = 3 ，则第 1 个人必须位于大小为 3 的组中
每个人应该恰好只出现在一个组中，并且每个人必须在一个组中。如果有多个答案，返回其中任何 一个。可以 保证 给定输入 至少有一个 有效的解。

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

先用哈希表计算组的个数，以及各个尺寸组的初始位置
然后把对应数的下标一一放入结果的对应数组中即可

class Solution {
public:
    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
        map<int,int> m;
        for(int num:groupSizes)
            m[num]++;
        int groups = 0;
        for(auto it=m.begin();it!=m.end();it++){
            int temp = groups;//记录当前尺寸组的初始位置
            groups = groups + it->second/it->first;
            it->second = temp;
        }
        vector<vector<int>> res(groups);//存储结果
        for(int i=0;i<groupSizes.size();i++){
            int index = m[groupSizes[i]];//组下标
            if(res[index].size()==groupSizes[i]) m[groupSizes[i]]++;//组满后移一个索引
            res[m[groupSizes[i]]].push_back(i);//放到对应组内
        }
        return res;
    }
};

标签：num,int,用户,second,vector,分组,groups,groupSizes,LeetCode
来源： https://www.cnblogs.com/929code/p/16578402.html