leetcode-166周赛-5280-用户分组

题目描述：

自己的提交：

class Solution:
    def groupThePeople(self, groupSizes: List[int]) -> List[List[int]]:
        ans = []
        dic = {}
        for v,i in enumerate(groupSizes):
            if i in dic:
                dic[i].append(v)
                if len(dic[i]) == i:
                    ans.append(dic[i])
                    dic.pop(i)
            else:
                dic[i] = [v]
                if i == 1:
                    ans.append(dic[i])
                    dic.pop(i)
        return ans

优化;

class Solution:
    def groupThePeople(self, g: List[int]) -> List[List[int]]:
        res = []
        d = collections.defaultdict(list)
        for i, x in enumerate(g):
            d[x].append(i)
            if len(d[x]) == x:
                res.append(d[x])
                d[x] = []
        return res