问题描述
有 n 个人被分成数量未知的组。每个人都被标记为一个从 0 到 n - 1 的唯一ID 。
给定一个整数数组 groupSizes ,其中 groupSizes[i] 是第 i 个人所在的组的大小。例如,如果 groupSizes[1] = 3 ,则第 1 个人必须位于大小为 3 的组中。
返回一个组列表,使每个人 i 都在一个大小为 groupSizes[i] 的组中。
每个人应该 恰好只 出现在 一个组 中,并且每个人必须在一个组中。如果有多个答案,返回其中 任何 一个。可以 保证 给定输入 至少有一个 有效的解。
示例 1:
输入: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]]。
示例 2:
输入:groupSizes = [2,1,3,3,3,2] 输出:[[1],[0,5],[2,3,4]]
提示:
- groupSizes.length == n
- 1 <= n <= 500
- 1 <= groupSizes[i] <= n
解题思路
在本题中,groupSizes[i]
代表第 i 个人所在的组的大小,处于同一个组的人,groupSizes[i]
一定相同,我们只需要通过Map来记录不同的Size,以及该Size内的人(下标)。
同时需要注意以下一点
- 当一个组满员的时候,该组就添加到结果中,并且Map重制该大小的组
AC代码
var groupThePeople = function(groupSizes) { const result = [] const map = new Map() for(let i = 0; i < groupSizes.length; i++){ if(!map.get(groupSizes[i])){ // 不存在该组 或者已经被置空 if(groupSizes[i] === 1){ // 一个人为一组则直接推入结果集 result.push([i]) }else{ map.set(groupSizes[i],[i]) } }else{ // 当前这个人加入该组的时候恰好满员,则推入结果并且置空该组 if(map.get(groupSizes[i]).length === groupSizes[i] - 1){ result.push([...map.get(groupSizes[i]),i]) map.set(groupSizes[i],false) }else{ map.set(groupSizes[i],[...map.get(groupSizes[i]),i]) } } } return result };