[프로그래머스] 체육복 Python

잘 푼거같은데 자꾸 테스트 통과를 못해서

 

엄청 찾다가

 

인덱스가 문제라는걸 알았다

 

항상 range 든 list 든 탐색할때 0 부터 시작한다는걸 잊지말자!!

def solution(n, lost, reserve):
    cloth = [1]*(n+1)
    for i in range(1,n+1):
        if i in lost and i in reserve:
            cloth[i] = 1
            continue
        if i in lost:
            cloth[i] = 0
        if i in reserve:
            cloth[i] = 2

    print(cloth)
    answer = 0
    for i in range(1, n+1):
        if cloth[i] == 0:
            if i > 1:
                if cloth[i - 1] == 2:
                    cloth[i] +=1
                    cloth[i - 1] -=1
                    continue
            if i < n:
                if cloth[i+1] ==2:
                    cloth[i] += 1
                    cloth[i+1] -=1
    for i in range(1, n+1):
        if cloth[i] > 0:
            answer+=1
    print(cloth)
    return answer