Update 0027-remove-element.py

Author: neetcode-gh

python/0027-remove-element.py

@@ -1,10 +1,8 @@
-class Solution(object):
-    def removeElement(self, nums, val):
-        """
-        :type nums: List[int]
-        :type val: int
-        :rtype: int
-        """
-        for i in range(nums.count(val)):  # loop how many val is in the list
-            nums.remove(val)  # remove each val one by one
-        return len(nums)  # return len of nums
+class Solution:
+    def removeElement(self, nums: List[int], val: int) -> int:
+        k = 0
+        for i in range(len(nums)):
+            if nums[i] != val:
+                nums[k] = nums[i]
+                k += 1
+        return k