有 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

 

思路：

使用Hashmap

以groupSize为key, 以ArrayList(ArrayList())为value, 这里存放的Integer就是userId(也就是数组index)

插入策略：

先判断是否有key，无key则直接生成新的ArrayList(ArrayList())， 并且添加一个子ArrayList, 同时插入userId

有key再判断取出的ArrayList内部的最后一个ArrayList长度是否等于对应的groupSize， 如果相等， 则追加到新组， 否则追加

题解：

public class Test {

    public static void main(String[] args) {

        System.out.println(groupThePeople(new int[]{3,3,3,3,1,3,3}));

    }

 

    private static HashMap resultMap = new HashMap<>();

 

    public static List> groupThePeople(int[] groupSizes) {

        ArrayList> results = new ArrayList<>();

    function(){ //点差 /tupian/20230521/ppspan >        for (int i=0;i

            add(i,groupSizes[i]);

        }

        for (ArrayList i : resultMap.values()) {

            results.addAll(i);

        }

        return results;

    }

 

    static void  add(int id, int groupSize) {

        if (resultMap.get(groupSize) == null) {

            ArrayList> groupList = new ArrayList<>();

            List group=new ArrayList<>();

            group.add(id);

            groupList.add(group);

            resultMap.put(groupSize, groupList);

        } else {

            ArrayList> a = resultMap.get(groupSize);

            if (a.get(a.size() - 1).size() == groupSize ) {

                List group=new ArrayList<>();

                group.add(id);

                a.add(group);

            } else {

                a.get(a.size() - 1).add(id);

            }

        }

    }

}

 

网站栏目：无尽算法之用户编组-创新互联
链接地址：http://scyanting.com/article/dppeih.html