二、用户分组（Weekly Contest 166）

最新推荐文章于 2021-09-20 16:45:04 发布

原创最新推荐文章于 2021-09-20 16:45:04 发布 · 106 阅读

0 ·

CC 4.0 BY-SA版权

竞赛专栏收录该内容

253 篇文章

订阅专栏

本文介绍了一种基于用户组大小进行用户分组的算法实现，通过遍历用户ID，根据其所属组大小进行分组，确保每组人数正确。算法采用ArrayList存储各组用户ID，最终返回所有组的集合。

题目描述：
有 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/group-the-people-given-the-group-size-they-belong-to
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

也是十分简单的一道题目：

class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        int len = groupSizes.length;
        boolean [] visited = new boolean[len];
        List<List<Integer>> result = new ArrayList<>();
        for (int i = 0; i < groupSizes.length; i++) {
            int gr = groupSizes[i];
            if(gr == 1){
                ArrayList<Integer> objects = new ArrayList<>();
                objects.add(i);
                visited[i] = true;
                result.add(objects);
            }else if(visited[i]){
                continue;
            }else{
                ArrayList<Integer> objects = new ArrayList<>();
                objects.add(i);
                visited[i] = true;
                for (int j = i; j < visited.length; j++) {
                    boolean b = visited[j];
                    if(b || groupSizes[i] != gr){
                        continue;
                    }
                    if(groupSizes[j] == gr){
                        visited[j] = true;
                        objects.add(j);
                    }
                    if(objects.size() == gr){
                        break;
                    }
                }
                              
                result.add(objects);

            }
        }
        return result; 
    }
}