Optimize solution to Sort Colors

soapyigu · soapyigu · commit 4ec4f29592cc · 2017-04-10T22:27:45.000-07:00
diff --git a/Sort/SortColors.swift b/Sort/SortColors.swift
@@ -6,13 +6,7 @@
 
 class SortColors {
     func sortColors(_ nums: inout [Int]) {
-        guard nums.count > 1 else {
-            return
-        }
-        
-        var red = 0
-        var blue = nums.count - 1
-        var i = 0
+        var red = 0, blue = nums.count - 1, i = 0
         
         while i <= blue {
             if nums[i] == 0 {
@@ -28,7 +22,7 @@ class SortColors {
         }
     }
     
-    private func _swap<T>(_ nums: inout [T], _ p: Int, _ q: Int) {
+    fileprivate func _swap<T>(_ nums: inout [T], _ p: Int, _ q: Int) {
         (nums[p], nums[q]) = (nums[q], nums[p])
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -6,13 +6,7 @@`
`6`	`6`
`7`	`7`	`class SortColors {`
`8`	`8`	`func sortColors(_ nums: inout [Int]) {`
`9`		`- guard nums.count > 1 else {`
`10`		`- return`
`11`		`- }`
`12`		`-`
`13`		`- var red = 0`
`14`		`- var blue = nums.count - 1`
`15`		`- var i = 0`
	`9`	`+ var red = 0, blue = nums.count - 1, i = 0`
`16`	`10`
`17`	`11`	`while i <= blue {`
`18`	`12`	`if nums[i] == 0 {`
`@@ -28,7 +22,7 @@ class SortColors {`
`28`	`22`	`}`
`29`	`23`	`}`
`30`	`24`
`31`		`- private func _swap<T>(_ nums: inout [T], _ p: Int, _ q: Int) {`
	`25`	`+ fileprivate func _swap<T>(_ nums: inout [T], _ p: Int, _ q: Int) {`
`32`	`26`	`(nums[p], nums[q]) = (nums[q], nums[p])`
`33`	`27`	`}`
`34`	`28`	`}`