携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第17天，点击查看活动详情

刷题的日常-2022年8月12日

一天一题，保持脑子清爽

用户分组

来自leetcode的 1282 题，题意如下：

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

示例如下：

输入：groupSizes = [3,3,3,3,3,1,3]
输出：[[5],[0,1,2],[3,4,6]]
解释：
第一组是 [5]，大小为 1，groupSizes[5] = 1。
第二组是 [0,1,2]，大小为 3，groupSizes[0] = groupSizes[1] = groupSizes[2] = 3。
第三组是 [3,4,6]，大小为 3，groupSizes[3] = groupSizes[4] = groupSizes[6] = 3。 
其他可能的解决方案有 [[2,1,6],[5],[0,4,3]] 和 [[5],[0,6,2],[4,3,1]]。

理解题意

我们可以从题意中提取的条件如下：

• 给定一个数组，需要给这个元素进行分组
• 每个元素代表分组的大小
• 我们元素放到对应大小的分组内

做题思路

其实很简单，题目已经保证了输入至少有一个有效解，那么我们可以在分组大小用Map存储起来，在遍历的过程中往里添加数据：

• 开辟一个Map用于保存分组大小的映射
• 遍历的时候如果没有改分组信息，新增一个
• 如果有的话，就将当前索引添加进去
• 如果大小已经满足，就其加到结果集当中

代码实现

代码实现如下：

public class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        List<List<Integer>> result = new LinkedList<>();
        Map<Integer, List<Integer>> sizeMap = new HashMap<>();

        for (int i = 0; i < groupSizes.length; i++) {
            int groupSize = groupSizes[i];
            List<Integer> list = sizeMap.computeIfAbsent(groupSize, o -> {
                ArrayList<Integer> tmp = new ArrayList<>(o);
                result.add(tmp);
                return tmp;
            });
            list.add(i);
            if (list.size() == groupSize) {
                sizeMap.remove(groupSize);
            }
        }

        return result;
    }
}