yan chuanli 2 years ago
parent
commit
f955a22c62
1 changed files with 22 additions and 0 deletions
  1. 22 0
      remove_element.py

+ 22 - 0
remove_element.py

@@ -0,0 +1,22 @@
+from typing import List
+import pytest
+
+def removeElement(nums: List[int], val: int):
+    i = 0
+    while i < len(nums):
+        if nums[i] == val:
+            nums.remove(nums[i])
+        else:
+            i += 1
+    return len(nums)
+
+
+@pytest.mark.parametrize(
+    "nums, val, expect",
+    [
+        ([-2, 3, 1, 5], 3, 3),
+        ([2, 3, 3, 2], 3, 2)
+    ]
+)
+def test_cases(nums, val, expect):
+    assert removeElement(nums, val) == expect