[Array] Refactor solution to Move Zeros

soapyigu · soapyigu · commit 06a0189c6331 · 2020-02-29T19:59:14.000-08:00
diff --git a/Array/MoveZeroes.swift b/Array/MoveZeroes.swift
@@ -8,18 +8,16 @@
 
 class MoveZeroes {
     func moveZeroes(_ nums: inout [Int]) {
-        var idx = 0
+        var nonZeroIdx = 0
         
-        for (i, num) in nums.enumerated() {
-            if num != 0 {
-                nums[idx] = num
-                idx += 1
-            }
+        for num in nums where num != 0 {
+            nums[nonZeroIdx] = num
+            nonZeroIdx += 1
         }
         
-        while idx < nums.count {
-            nums[idx] = 0
-            idx += 1
+        while nonZeroIdx < nums.count {
+            nums[nonZeroIdx] = 0
+            nonZeroIdx += 1
         }
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -8,18 +8,16 @@`
`8`	`8`
`9`	`9`	`class MoveZeroes {`
`10`	`10`	`func moveZeroes(_ nums: inout [Int]) {`
`11`		`- var idx = 0`
	`11`	`+ var nonZeroIdx = 0`
`12`	`12`
`13`		`- for (i, num) in nums.enumerated() {`
`14`		`- if num != 0 {`
`15`		`- nums[idx] = num`
`16`		`- idx += 1`
`17`		`- }`
	`13`	`+ for num in nums where num != 0 {`
	`14`	`+ nums[nonZeroIdx] = num`
	`15`	`+ nonZeroIdx += 1`
`18`	`16`	`}`
`19`	`17`
`20`		`- while idx < nums.count {`
`21`		`- nums[idx] = 0`
`22`		`- idx += 1`
	`18`	`+ while nonZeroIdx < nums.count {`
	`19`	`+ nums[nonZeroIdx] = 0`
	`20`	`+ nonZeroIdx += 1`
`23`	`21`	`}`
`24`	`22`	`}`
`25`	`23`	`}`