每日一题:1282. 用户分组

81 阅读3分钟

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

今天星期五可惜KFC活动过去了,昨天加班到7点没得吃肯德基,要加到9点才有,是你你吃不吃? 今天带来的是一道leetcode的每日一题

前言

[1417. 用户分组]【中等题】,打开每日一题,嗯?通过率80多?这不简单题吗。

审题

有 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]]。

首先,先给大家讲一下这个题目怎么回事,这题目真的太狗了,导致我跟我的同事看了半天才理解过来。题目的意思是数组分组,其实这道题是要我们将数组中值相等的索引放到一个分组中,数组的索引对应的是用户的 ID,用上面的代码来举例。例如用户 0 所在组大小为 2,也就是说这个组有 2 个人。然后索引 5,也就是用户 5 所在的用户组也是 2。说明用户 0 和用户 5 是在一个组。索引 1,也就是用户 1 所在的组为 1,只有它自己。用户 2 3 4 为一组,因为组大小为 3,有 3 个人。假设我举一个例子 11 22 3333 这个数组,他不成立需要返回空数组,原因如下:1. 因为数字的长度不等于所对应的数字,所以直接pass掉。那么我们理解了题目,我们就来开始做题把。

我的做题思路一开始做错了,我一开始想着是先弄个map存放每一个数字的存放次数,但是到最后发现对比数字的次数之后还要进行数字的下标匹配,搞得我一度想用 StringBuilder去做了,直接把多余的拼起来假设 333333 这个是可以分为两组的,我初始想法便是key:31 value:333/key:32,value:333显而易见这很蠢,所以不如采用下面的方法因为我们上面也分析了说了长度与数字相等,那么我们的map直接存储数字的长度即可。

编码

class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        Map<Integer, List<Integer>> groups = new HashMap<Integer, List<Integer>>();
        int n = groupSizes.length;
        for (int i = 0; i < n; i++) {
            int size = groupSizes[i];
            groups.putIfAbsent(size, new ArrayList<Integer>());
            groups.get(size).add(i);
        }
        List<List<Integer>> groupList = new ArrayList<List<Integer>>();
        for (Map.Entry<Integer, List<Integer>> entry : groups.entrySet()) {
            int size = entry.getKey();
            List<Integer> people = entry.getValue();
            int groupCount = people.size() / size;
            for (int i = 0; i < groupCount; i++) {
                List<Integer> group = new ArrayList<Integer>();
                int start = i * size;
                for (int j = 0; j < size; j++) {
                    group.add(people.get(start + j));
                }
                groupList.add(group);
            }
        }
        return groupList;
    }
}

耗时

image.png