携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第17天,点击查看活动详情
说在前面
🎈不知道大家对于算法的学习是一个怎样的心态呢?为了面试还是因为兴趣?不管是处于什么原因,算法学习需要持续保持,今天让我们一起来看看这一道题目————
用户分组
。
题目描述
有 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
思路分析
刚开始读完题目可能还有点不知道题目要我们做什么,所以我们可以结合例子来对题目进行理解。题目的要求是这样的,主要可以注意以下几点:
- 有n个人,ID 为 0 到 n - 1
- groupSizes[i] 是第 i 个人所在的组的大小
- 每个人只能出现在一个组中
简单来说就是我们要对n个人进行分组,使其满足第i个人所在的组中总共有groupSizes[i]个人,那么我们可以这样做:
- 1、使用一个hash表记录当前分组容量为n且组内人数未满的组的下标;
- 2、遍历判断当前是否存在分组容量为
groupSizes[i]
且组内人数未满的组; - 3、不存在则新建一个组,并更新hash表中对应的值;
let ind = map[groupSizes[i]];
if(ind === void 0 || res[ind].length === groupSizes[i]){
map[groupSizes[i]] = res.length;
res.push([i]);
}
- 4、存在则将当前成员放入组内
else{
res[ind].push(i);
}
AC代码
/**
* @param {number[]} groupSizes
* @return {number[][]}
*/
var groupThePeople = function(groupSizes) {
const res = [],map = {};
for(let i = 0; i < groupSizes.length; i++){
let ind = map[groupSizes[i]];
if(ind === void 0 || res[ind].length === groupSizes[i]){
map[groupSizes[i]] = res.length;
res.push([i]);
}else{
res[ind].push(i);
}
}
return res;
};
说在后面
🎉这里是JYeontu,喜欢算法,GDCPC打过卡;热爱羽毛球,大运会打过酱油。毕业一年,两年前端开发经验,目前担任H5前端开发,算法业余爱好者,有空会刷刷算法题,平时喜欢打打羽毛球🏸 ,也喜欢写些东西,既为自己记录📋,也希望可以对大家有那么一丢丢的帮助,写的不好望多多谅解🙇,写错的地方望指出,定会认真改进😊,在此谢谢大家的支持,我们下文再见🙌。