-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathgroup_people.py
More file actions
30 lines (25 loc) · 930 Bytes
/
group_people.py
File metadata and controls
30 lines (25 loc) · 930 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
class Solution:
def groupThePeople(self, groupSizes: List[int]) -> List[List[int]]:
def split_group(group, size):
groups = []
curr_group = []
for i, p in enumerate(group):
if i % size == 0:
if curr_group:
groups.append(curr_group)
curr_group = [p]
else:
curr_group.append(p)
groups.append(curr_group)
return groups
groups = defaultdict(list)
split_groups = []
for p, group_size in enumerate(groupSizes):
groups[group_size].append(p)
for size, group in groups.items():
if len(group) > size:
for g in split_group(group, size):
split_groups.append(g)
else:
split_groups.append(group)
return split_groups