Leetcode - Remove Element

15 Oct 2024 in Algorithm

알고리즘 공부

Daily Leetcode

• 🥽 Solution
  • 👻 Complexity

문제 링크

특정 원소를 지우는 문제이다. 이건 사실 Naive를 생각하는게 더 어려운 문제이다.

핵심은 in-place 에서 삭제가 이뤄져야 한다. 그럼 결국 포인터를 쓸 수 밖에..

🥽 Solution

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:

        n = len(nums)
        idx = 0

        for i in range(n) :
            if nums[i] != val:
                nums[idx] = nums[i]
                idx += 1
        
        return idx
            

지워도 되지 않은 원소라면 앞에서부터 채워넣어준다. 그게 아니라면 그냥 무시하고 넘어가면서 idx를 채워 주기만 하면 된다.

👻 Complexity

• Time : O(n)
• Space : O(1)