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

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

코드

def solution(n, lost, reserve):
    lost2 = [i for i in lost if not i in reserve]
    reserve2 = [i for i in reserve if not i in lost]
    answer = n - len(lost2)
    
    for i in lost2:    
        if i-1 in reserve2:
            reserve2.remove(i-1)
            answer+=1
        elif i+1 in reserve2:
            reserve2.remove(i+1)
            answer+=1
    return answer

여분이 없으면서 잊어버린 index만 lost에 남긴다.

체육복이 2개 있는 index만 reserve에 남긴다.

양 옆 중에 reserve에 있는 값이 있다면 answer에 1을 더한다.

Discussion and feedback