leetcode1282

 1 class Solution:
 2     def groupThePeople(self, groupSizes: List[int]) -> List[List[int]]:
 3         dic = {}
 4         n = len(groupSizes)
 5         for i in range(n):
 6             if groupSizes[i] not in dic:
 7                 dic[groupSizes[i]] = [i]
 8             else:
 9                 dic[groupSizes[i]].append(i)
10         res = []
11         for k,v in dic.items():
12             #k:每组几个，v集合
13             temp = []
14             for j in range(len(v)):
15                 temp.append(v[j])
16                 if len(temp) == k:
17                     res.append(temp[:])
18                     temp.clear()
19         return res