Leetcode Python - 27. Remove Element

Leetcode #27 - Remove Element

이번에는 특정 챕터별로 문제들을 풀어보겠습니다. string을 집중적으로 풀어보겠습니다.

리트코드의 문제 27 ‘Remove Element’을 파이썬으로 풀어 보도록 하겠습니다.

26번 문제와 유사하게 풀 수 있습니다.

전체 코드는 아래와 같습니다.

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        if not nums:
            return 0
        
        idx = 0
        while idx < len(nums):
            if nums[idx] == val:
                nums.pop(idx)
            else:
                idx += 1
        
        return len(nums)

시간복잡도는 O(n) : while문(n) 공간복잡도는 O(1) : 상수 idx 선언