用户分组

1282. 用户分组

难度中等44收藏分享切换为英文接收动态反馈

有 n 位用户参加活动，他们的 ID 从 0 到 n - 1，每位用户都 恰好 属于某一用户组。给你一个长度为 n 的数组 groupSizes，其中包含每位用户所处的用户组的大小，请你返回用户分组情况（存在的用户组以及每个组中用户的 ID）。

你可以任何顺序返回解决方案，ID 的顺序也不受限制。此外，题目给出的数据保证至少存在一种解决方案。

示例 1：

输入：groupSizes = [3,3,3,3,3,1,3]
输出：[[5],[0,1,2],[3,4,6]]
解释： 
其他可能的解决方案有 [[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

小根堆

• 先对数字排序，排序的时候要保留$index$
• 从小到大选择数字的个数

class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return Integer.compare(o1[0], o2[0]);
            }
        });

        for (int i = 0; i < groupSizes.length; i++) {
            priorityQueue.offer(new int[]{groupSizes[i], i});
        }

        List<List<Integer>> res = new ArrayList<>();
        while (!priorityQueue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
            int size = priorityQueue.peek()[0];
            for (int i = 0; i < size; i++) {
                list.add(priorityQueue.poll()[1]);
            }
            res.add(list);
        }
        return res;
    }
}

时间复杂度$O(nlog(n))$,空间复杂度$O(n)$

小根堆

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Objects;
import java.util.PriorityQueue;

public class Solution {

    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        int len = groupSizes.length;
        List<List<Integer>> res = new ArrayList<>();

        // 特判
        if (len == 0) {
            return res;
        }

        // 默认就是最小堆，int[] 的长度为 2，第 1 个数是数组元素，第 2 个数是数组索引
        // 其实就是把数和索引绑定在了一起，在堆中参与比较的是数，索引是依附于数的
        PriorityQueue<int[]> minHeap = new PriorityQueue<>(len, Comparator.comparingInt(o -> o[0]));
        for (int i = 0; i < len; i++) {
            minHeap.add(new int[]{groupSizes[i], i});
        }

        while (!minHeap.isEmpty()) {
            int curSize = minHeap.peek()[0];

            List<Integer> current = new ArrayList<>();
            for (int i = 0; i < curSize; i++) {
                current.add(Objects.requireNonNull(minHeap.poll())[1]);
            }

            res.add(current);
        }
        return res;
    }
}

作者：liweiwei1419
链接：https://leetcode-cn.com/problems/group-the-people-given-the-group-size-they-belong-to/solution/you-xian-dui-lie-by-liweiwei1419/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

赏

谢谢你请我吃糖果

微信