LeetCode刷题笔记：1282.用户分组

1. 问题描述

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

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

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

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

2. 解题思路

将数组 groupSizes 的元素按照其大小分配到对应大小的数组中。
维护一个 HashMap，用于存放数组元素 groupSizes[i] 及其对应的 List。

• 如果哈希表中没有该元素，则将该元素和一个空 ArrayList 加入到该哈希表中，并向空列表添加数组元素的下标 i。
• 判断哈希表中 key 对应的 value 即列表是否已填充满，若已填充满则将该列表添加到结果集中，并从哈希表中删除该键值对。

3. 代码实现

class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        int n = groupSizes.length;
        List<List<Integer>> res = new ArrayList<>();
        HashMap<Integer, List<Integer>> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            int key = groupSizes[i];
            if(!map.containsKey(key)) {
                map.put(key, new ArrayList<Integer>());
            } 
            map.get(key).add(i);
            if (map.get(key).size() == key) {
                res.add(map.get(key));
                map.remove(key);
            }
        }
        return res;
    }
}