From 8547e55f43fbdc77fb327ebac51ad612f99bee13 Mon Sep 17 00:00:00 2001 From: Sandy Date: Mon, 10 Dec 2018 16:48:38 +0800 Subject: [PATCH] Update remove_element.go --- remove_element/remove_element.go | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/remove_element/remove_element.go b/remove_element/remove_element.go index 1765abf..408ac94 100644 --- a/remove_element/remove_element.go +++ b/remove_element/remove_element.go @@ -1,13 +1,13 @@ package remove_element func removeElement(nums []int, val int) int { - start := 0 - for i := 0; i < len(nums); i++ { - if nums[i] == val { - continue - } - nums[start] = nums[i] - start++ - } - return start + l := 0 + for _, v := range nums { + if v != val { + nums[l] = v + l++ + } + } + + return l }