[杨小白]_leetcode_1282. 用户分组

69 阅读2分钟

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

前言

小白算法比较菜,希望能激励我每日更新,从leetcode第一题开始,2022年目标300题,记录从0到1的全过程!!

1282. 用户分组

2022年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

题目解析

写题干的人出来挨打~写的真绕,看了好半天才看懂到底说的说什么。

一共有n个人,每个人的编号id唯一从0开始到n-1,对应了题干给的groupSizes数组,groupSizes[0]就是代表id为0的人,假设groupSizes[0]的值为3,表示id为0的人要分在一个组,这个租只有三个人。

拿题目给的示例,groupSizes = [3,3,3,3,3,1,3],id为0,1,2,3,4,6的人要分在只有三个人的组里,id为5的人要分在一个只有一个人的组里,所以id为5的人单独为一组[5],剩余的6个人随便平均分为两组就可以了[0,1,2],[3,4,6]。所以结果就是[[5],[0,1,2],[3,4,6]]

示例2种,groupSizes = [2,1,3,3,3,2],id为0和5的人 要分在有两个人的组里,所以他俩一组[0,5];id为1的人要分在只有一个人的组里,所以他自己一组[1];剩下三个人的值为3,要分在有三个人的组里,所有又有一个组[2,3,4]。所以最后的结果就是[[1],[0,5],[2,3,4]]。

2.解法

class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        HashMap<Integer, ArrayList<Integer>> map = new HashMap<>();
        for(int i = 0; i < groupSizes.length; i++) {
            int key = groupSizes[i];
            if(map.containsKey(key)) {
                map.get(key).add(i);
            } else {
                ArrayList<Integer> arr = new ArrayList();
                arr.add(i);
                map.put(key,arr);
            }
        }
        List<List<Integer>> res = new ArrayList<>();
        for(Map.Entry<Integer, ArrayList<Integer>> entry : map.entrySet()) {
            ArrayList<Integer> list = entry.getValue();
            ArrayList<Integer> cur = new ArrayList<>();
            for(int i = 0; i < list.size(); i++) {
                cur.add(list.get(i));
                if(cur.size()==entry.getKey()) {
                    res.add(cur);
                    cur = new ArrayList<>();
                }
            }
        }
        return res;
    }
}

提交排名

image.png

这个题的题目有点难理解,但是做起来的思路并不复杂,遍历数组,用一个map存下所有的用户,key为他们想要分为几人一组,value是一个list,记录了这些人的id。比如[3,3,3,3,3,1,3],那么map就有两个值,1:5和3:[0,1,2,3,4,6]。

存入map后,从头遍历,拿出key和对应的list,将list每key个分为一组,压入返回的res列表中。遍历完整个map,res也就是最终的结果了。

3.结束

每日一题每日一题,gogogo,前进的道路还很远啊!!!