프로그래머스 체육복

문제

def solution(n, lost, reserve):
    # 잃어버린 학생과 여분의 체육복을 가진 학생을 정렬
    lost.sort()
    reserve.sort()

    # 여분의 체육복이 있는 학생이 잃어버린 학생이라면, 빌려주기
    for i in reserve[:]:
        if i in lost:
            reserve.remove(i)
            lost.remove(i)

    # 여분의 체육복을 가진 학생이 다른 잃어버린 학생에게 빌려주기
    for i in reserve:
        if i-1 in lost:
            lost.remove(i-1)
        elif i+1 in lost:
            lost.remove(i+1)
    
    # 체육복을 받을 수 없는 학생을 제외한 나머지 학생 수 반환
    return n - len(lost)