1282. Group the People Given the Group Size They Belong To

donutloop · donutloop · commit 21fb8ae93ea0 · 2024-03-16T14:28:07.000+01:00
diff --git a/Group the People Given the Group Size They Belong To/kata.go b/Group the People Given the Group Size They Belong To/kata.go
@@ -0,0 +1,21 @@
+package kata
+
+func groupThePeople(groupSizes []int) [][]int {
+	buckets := make(map[int][]int, 0)
+	dividedGroups := make([][]int, 0)
+	for i, groupSize := range groupSizes {
+		bucket := buckets[groupSize]
+		if bucket == nil {
+			bucket = make([]int, 0)
+		}
+
+		bucket = append(bucket, i)
+		if len(bucket) == groupSize {
+			dividedGroups = append(dividedGroups, bucket)
+			buckets[groupSize] = nil
+		} else {
+			buckets[groupSize] = bucket
+		}
+	}
+	return dividedGroups
+}
