携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第12天,点击查看活动详情
前言
今天的题目为中等,但是只是一道简单的哈希表应用题,是用哈希表纪录下状态,很容易就能够完成题目。
每日一题
今天的题目是 1282. 用户分组,难度为中等
-
有 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
题解
哈希表
简单题目,通过哈希表先进行一次循环,保存下当前循环的时候,每一个下标对应的个数是多少,一次存入一个哈希表,将相等个数的下标归为一类,这样就能够获得每一个下标都需要拼凑到多长的数组中去。
然后再次循环哈希表,每次循环到的会是一个长度对应一个数组,将这个数组按照长度进行分割,每一个长度为一组,最后加入答案数组当中,循环完哈希表之后,就也能够得到答案了。
我们拿 [3,3,3,3,3,1,3] 举个例子,第一次循环用长度作为key,下标作为元素,形成哈希表。
[3,3,3,3,3,1,3] => Map(2) { 3 => [ 0, 1, 2, 3, 4, 6 ], 1 => [ 5 ] }
然后通过循环这个哈希表,第一次会拿到 3 和一个 数组 [ 0, 1, 2, 3, 4, 6 ] 通过 3 将数组分割为长度为 3 的等份,第二次拿到的操作也是一样的,最后统一加入一个答案数组当中。
3 => [ 0, 1, 2, 3, 4, 6 ] => [ 0, 1, 2 ],[ 3, 4, 6 ]
function groupThePeople(groupSizes: number[]): number[][] {
const map = new Map<number, number[]>();
const n = groupSizes.length;
for (let i = 0; i < n; i++) {
const size = groupSizes[i];
if (!map.has(size)) {
map.set(size, []);
}
map.get(size).push(i);
}
const res: number[][] = [];
for (const [size, people] of map.entries()) {
for (let i = 0; i < people.length; i += size) {
res.push(people.slice(i, i + size));
}
}
return res;
}