用户分组

42 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 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前端开发,算法业余爱好者,有空会刷刷算法题,平时喜欢打打羽毛球🏸 ,也喜欢写些东西,既为自己记录📋,也希望可以对大家有那么一丢丢的帮助,写的不好望多多谅解🙇,写错的地方望指出,定会认真改进😊,在此谢谢大家的支持,我们下文再见🙌。