[Python] 반복문에서 remove 사용시 유의할 점

240513 Today I Learn

문제상황

 

프로그래머스

def solution(nums):
    answer = []
    
    for i in range(0, len(nums)):
        for j in range(i+1, len(nums)):
            for k in range(j+1, len(nums)):
                answer.append(nums[i]+nums[j]+nums[k])
                
    for n in answer:
        cnt = []
        for m in range(2,n):
            if n % m == 0:
                cnt.append(m)
        if len(cnt)!=0:
            answer.remove(n) 

    return len(answer)

 

🎓 로직

• 첫번째 for문에서 입력받은 숫자 리스트 nums에 있는 숫자 중 3개를 뽑아 더하고 이를 answer 리스트에 append해준다.
  → 이제 answer 리스트는 nums에서 랜덤으로 뽑은 서로 다른 세개의 숫자를 더한 리스트가 된다.
• 두번째 for문은 소수를 판정하기 위한 반복문이다.
  • 소수란 1과 자기 자신만을 약수로 하는 수로, 임의의 숫자 x가 2와 x-1 사이에 약수를 가진다면 x는 약수가 될 수 없다.
  • 따라서 2부터 n-1까지의 숫자를 m으로 받는 반복문을 작성한 뒤 그것이 answer에 있는 n의 약수라면 빈 리스트 cnt에 저장한다.
    → answer의 모든 원소 n은 cnt를 하나씩 갖는다.
  • 만약 cnt의 길이가 0이라면 소수이고, 0이 아니라면 소수가 아니다. 따라서 answer에서 소수가 아닌 n을 remove를 이용해 삭제한다.
• 마지막으로 answer에는 소수가 아닌 모든 n이 삭제되어 소수인 수만 남으므로 answer의 길이를 len을 이용해 구해준다.

🔍 에러탐색

문제가 발생하는 코드를 탐색하기 위해 반복문별로 코드를 분절해서 돌려보았다. 

두번째 코드 블럭까지는 의도에 맞게 nums에서 3가지의 수를 뽑아서 더한 10개의 숫자(5C3 = 10)가 출력된다.

그렇다면 문제가 생기는 코드 블럭은 3번째 코드블럭이다. 

 

for n in answer:
    cnt  = []
    for m in range(2,n):
        if n%m == 0:
            cnt.append(m)
    print(f"cnt: {cnt}")
    if len(cnt) !=0:
        answer.remove(n)
    print(f"answer: {answer}")

3번째 코드블럭에서 중간중간 print문을 넣어가며 어느 부분에서 에러가 발생하는지 확인해보자.

두번째 코드블럭의 output인 answer는 다음과 같다. [10, 9, 7, 14, 12, 11, 15, 13, 12, 17]

그러나 위 이미지와 같이 실제로 출력된 데이터를 보면

빨간색으로 표시한 다음의 숫자들이 빠진 채 코드가 돌아간 것을 확인할 수 있다.

[10, 9, 7, 14, 12, 11, 15, 13, 12, 17]

 

for n in answer:
    cnt  = []
    for m in range(2,n):
        if n%m == 0:
            cnt.append(m)
    print(f"cnt: {cnt}")

for문만 돌린 채 cnt를 출력해보면 이렇게 정상적으로 answer의 모든 원소들의 약수(2~n-1 사이)가 출력되는 것을 확인할 수 있다.

 

결국 remove로 인해 원 리스트인 answer에 영향이 가고,

그 영향으로 인해 문제가 발생하는 것이라고 예측할 수 있다.

 

list에서 remove를 사용하여 아이템을 제거하면서 동시에 for loop을 사용하는 경우, 
리스트의 길이가 바뀌기 때문에 인덱스가 망가질 수 있다.

for n in answer:
    print(f"answer: {answer}")
    cnt  = []
    for m in range(2,n):
        if n%m == 0:
            cnt.append(m)
    if len(cnt) >0:
        answer.remove(n)

remove되고 다음 n을 뽑을 때 answer 리스트는 다음과 같은 형태이다.

이처럼 remove를 사용하는 경우, 리스트의 인덱스가 변경되고

remove를 통해 제거된 10이 있던 answer[0] 자리에

원래 answer[1] 값이었던 9가 들어가면서 배열이 틀어지게 되는 것이다.

이 때문에 9, 12, 13, 17을 건너뛴 채 for문이 돌아가는 것..!

 

 

🕵️‍♀️ 문제해결

이 문제는 다음과 같이 sorted(answer)를 사용해 answer를 정렬함으로써 해결할 수 있다.

for n in sorted(answer):
    cnt  = []
    for m in range(2,n):
        if n%m == 0:
            cnt.append(m)
    print(f"cnt: {cnt}")
    if len(cnt) != 0:
        answer.remove(n)
    print(f"answer: {answer}")

 

튜터님의 조언
for문을 더 효율적으로 사용할 수 있는 로직을 생각해보자.

리스트에 무엇인가를 넣었다가 빼는 과정 자체가 비효율적이기 때문에

리스트 컴프리핸션을 이용해서 필요한 데이터만 리스트에 넣는 작업을 하는 것을 추천.