코딩 공부/Leetcode

[Python] 27. Remove Element

일하는 공학도 2025. 1. 18. 14:11
728x90

난이도 : Easy

 

nums라는 리스트에서 val 값을 전부 제거하고, 리스트에 남은 요소가 몇 개인지 return하는 문제

단, nums만 수정할 것

 

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        a = 0
        for i in range(len(nums)):
            if nums[i] != val:
                nums[a] = nums[i]
                a += 1
        return a
  1. for문에서 요소 하나씩 val와 같은지 확인
  2. val와 다른 요소들은 하나씩 자리를 바꿔버리고, 바뀐 수 a를 return

Runtime : 0ms

Memory : 17.91MB(14.24%)

728x90