From 1c79f8375ff2c982bc2009834d067343b29053be Mon Sep 17 00:00:00 2001 From: ivan Date: Wed, 6 Nov 2024 04:09:19 -0600 Subject: [PATCH] adding remove element algo --- .../top_150_questions_round_11/remove_element.py | 10 ++++++++++ .../test_remove_element_round_11.py | 11 +++++++++++ 2 files changed, 21 insertions(+) create mode 100644 src/my_project/interviews/top_150_questions_round_11/remove_element.py create mode 100644 tests/test_150_questions_round_11/test_remove_element_round_11.py diff --git a/src/my_project/interviews/top_150_questions_round_11/remove_element.py b/src/my_project/interviews/top_150_questions_round_11/remove_element.py new file mode 100644 index 00000000..4d6b0758 --- /dev/null +++ b/src/my_project/interviews/top_150_questions_round_11/remove_element.py @@ -0,0 +1,10 @@ +from typing import List, Union, Collection, Mapping, Optional +from abc import ABC, abstractmethod + +class Solution: + def removeElement(self, nums: List[int], val: int) -> int: + + while val in nums: + nums.remove(val) + + return len(nums) \ No newline at end of file diff --git a/tests/test_150_questions_round_11/test_remove_element_round_11.py b/tests/test_150_questions_round_11/test_remove_element_round_11.py new file mode 100644 index 00000000..d8bb982a --- /dev/null +++ b/tests/test_150_questions_round_11/test_remove_element_round_11.py @@ -0,0 +1,11 @@ +import unittest +from src.my_project.interviews.top_150_questions_round_11\ +.remove_element import Solution + +class RemoveElementTestCase(unittest.TestCase): + + def test_remove_element(self): + solution = Solution() + output = solution.removeElement(nums=[3,2,2,3], val=3) + target = 2 + self.assertEqual(output, target) \ No newline at end of file