HelloCho

[Python] #27. Remove Element(Easy) 본문

코테/LeetCode

[Python] #27. Remove Element(Easy)

choo2969 2020. 9. 6. 02:55

문제 

이 문제는 배열과 값이 주어질 때, 배열 안에서 주어진 값을 제거하고 값을 in-place방식으로 return 하는 문제이다. 

해결 방법:

for문을 통해서 val과 일치하는 값을 제거했다. del을 통해서 제거할 때 index값이 하나씩 당겨지기 때문에 s라는 변수를 통해서 index값을 맞춰줬다. 

class Solution:
    def removeElement(self, nums, val):
        s = 0
        for idx in range(len(nums)):
            if nums[idx-s] == val:
                del nums[idx-s]
                s +=1
        return len(nums)

결과 

28ms의 속도와 13.9MB로 통과했다! 

Comments