Create: 0075-sort-colors.swift

purnawirman · Feb 4, 2023 · 02a091a · 02a091a
1 parent 7bed4f9
commit 02a091a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/swift/0075-sort-colors.swift b/swift/0075-sort-colors.swift
@@ -0,0 +1,17 @@
+// Bucket Sort Solution
+class Solution {
+    func sortColors(_ nums: inout [Int]) {
+        var counts = [0, 0, 0]
+        for n in nums {
+            counts[n] += 1
+        }
+
+        var i = 0
+        for n in 0..<counts.count {
+            for j in 0..<counts[n] {
+                nums[i] = n
+                i += 1
+            }
+        }
+    }
+}