[Array] refactor code style and simplify solution

Lafree317 · May 4, 2016 · 24d16e9 · 24d16e9
1 parent 21886bd
commit 24d16e9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/Array/MoveZeroes.swift b/Array/MoveZeroes.swift
@@ -9,15 +9,12 @@
 class MoveZeroes {
     func moveZeroes(inout nums: [Int]) {
         var zeroIndex = 0
-        var left = 0
-        let right = nums.count
 
-        while left < right {
-            if nums[left] != 0 {
-                _swap(&nums, zeroIndex, left)
+        for i in 0 ..< nums.count {
+            if nums[i] != 0 {
+                _swap(&nums, zeroIndex, i)
                 zeroIndex += 1
             }
-            left += 1
         }
     }
 

diff --git a/Array/RemoveElement.swift b/Array/RemoveElement.swift
@@ -9,11 +9,7 @@ class RemoveElement {
     func removeElement(inout nums: [Int], _ val: Int) -> Int {
         var lastIndex = 0
 
-        if nums.count == 0 {
-            return lastIndex
-        }
-
-        for i in 0...nums.count - 1 {
+        for i in 0 ..< nums.count {
             if nums[i] != val {
                 nums[lastIndex] = nums[i]
                 lastIndex += 1