Leetcode:1282. 用户分组(贪心，hash）

最新推荐文章于 2022-08-14 10:38:50 发布

原创最新推荐文章于 2022-08-14 10:38:50 发布 · 268 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #数据结构 #算法 #c++

leetcode 专栏收录该内容

115 篇文章

订阅专栏

本文介绍了一种基于用户组大小分配用户至不同组别的算法实现。通过两步分解的方法，首先将用户按用户组大小分类，然后进一步细分为具体小组。此算法确保了每位用户都能被正确地分配到其所属的用户组。

1282. 用户分组

有 n 位用户参加活动，他们的 ID 从 0 到 n - 1，每位用户都恰好属于某一用户组。给你一个长度为 n 的数组 groupSizes，其中包含每位用户所处的用户组的大小，请你返回用户分组情况（存在的用户组以及每个组中用户的 ID）。

你可以任何顺序返回解决方案，ID 的顺序也不受限制。此外，题目给出的数据保证至少存在一种解决方案。

示例 1：

输入：groupSizes = [3,3,3,3,3,1,3]
输出：[[5],[0,1,2],[3,4,6]]
解释：
其他可能的解决方案有 [[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

这道题还是贪心的思想个人在于，我们首先将用户组相同的用户存在一区，然后每次拿出来一个用户组大小，显而易见这是必然能全部拿走的，所以每次贪心的拿走我们需要的就OK了，因此这里需要两次分解

第一次先把用户大概的分到相同用户组里，第二次再在用户组里分出相等的分量，问题就解决了

class Solution {
public:
    vector<vector<int>> groupThePeople(vector<int>& groupSizes) {
        unordered_map<int,vector<int>>groups;
        for(int i=0;i<groupSizes.size();i++){
            groups[groupSizes[i]].push_back(i);
        }
        vector<vector<int>>ans;
        for(auto group=groups.begin();group!=groups.end();group++){
            const int& gsize=group->first;
            vector<int>&users=group->second;
            for(auto iter=users.begin();iter!=users.end();iter=next(iter,gsize)){
                vector<int>dummy(iter,next(iter,gsize));
                ans.push_back(dummy);
            }
        }
        return ans;
    }
};