LeetCode #27

Kiyong Lee·2022년 1월 6일
0

leetcode

목록 보기
5/20

27. Remove Element


1. 코드

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        
        while nums.count(val) :
            nums.remove(val)
        
        return len(nums)

2. 풀이

첫 번째로 for문을 쓰려고 했는데, 삭제하면 리스트의 길이가 바뀌어서 list out of range가 발생해서 못하게 됨

두 번째로 while을 써야 하는데 처음에 생각한 while문의 첫 조건은 while idx < len(nums)였음

그런데 이렇게되면 마지막 요소에 대해 체크할 수가 없어 이것도 안됨

마지막으로 생각한 게 삭제할 요소의 갯수를 조건으로 설정해서 다 삭제되면 끝날 수 있도록 설정

profile
ISTJ인 K-개발자

0개의 댓글