[프로그래머스] 체육복

def solution(n, lost, reserve):

    student = []

    for i in range(1, n + 1):
        a =+ i
        student.append(a) #학생 리스트
        if i in lost:
            student.remove(i) #잃어버린 학생 모두 제거

    for i in reserve:
        front = i-1 # 앞 친구
        back = i+1 # 뒤 뒷친구
        if i in lost:
            student.append(i) # 여분의 옷 가지고 있는 학생이 잃어버렸을 때 추가
            lost.remove(i) # 잃어버린 학생에서 제거
        elif front in lost:
            student.append(front) #앞 친구 추가
        elif back in lost:
            student.append(back) #뒷 친구 추가

    student.sort()
    set_s = set(student)
    print(len(set_s))

    return len(set_s)