[99클럽] 27일차 TIL

https://leetcode.com/problems/group-the-people-given-the-group-size-they-belong-to/

 

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