1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
def solution(n, lost, reserve):
    answer = 0

    # n명의 학생 : 체육복 1개씩 있음
    students = [1 for x in range(n)]

    # lost일 경우 체육복 1개 차감
    for stud_id in lost:
        students[stud_id - 1] -= 1  # ----> (stud_id - 1)인 이유는 인덱스이기 때문에

    # reserve일 경우 체육복 1개 추가
    for stud_id in reserve:
        students[stud_id - 1] += 1  # ----> (stud_id - 1)인 이유는 인덱스이기 때문에

    # 앞번호의 학생부터 체육복 0개일 경우 주변 학생으로부터 가져옴
    for i in range(len(students)):
        if students[i] == 0:
            try:
                if students[i - 1] == 2:
                    students[i] += 1
                    students[i - 1] -= 1
                elif students[i + 1] == 2:
                    students[i] += 1
                    students[i + 1] -= 1
            except IndexError:
                continue

    # 체육복을 1개 이상 가지고 있는 학생의 수
    cnt = 0
    for stud in students:
        if stud >= 1:
            cnt += 1

    answer = cnt

    return answer