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
def solution(n, lost, reserve):
    answer = 0

    stu_arr = [1 for _ in range(n)]

    # 여분의 체육복이 있는 학생 초기화
    for i in range(len(reserve)):
        stu_arr[reserve[i]-1] +=1

    # 체육복을 잃어버린 학생 초기화
    for i in range(len(lost)):
        stu_arr[lost[i]-1] -= 1


    # 뒤에서 앞으로 체육복 빌려줌
    for i in range(n - 1, 0, - 1):
        if stu_arr[i] == 2:
            if stu_arr[i-1] == 0:
                stu_arr[i] = stu_arr[i-1] = 1

    # 앞에서 뒤로 체육복 빌려줌
    for i in range(n-1):
        if stu_arr[i] == 2:
            if stu_arr[i+1] == 0:
                stu_arr[i] = stu_arr[i+1] = 1



    print(stu_arr)
    for i in stu_arr:
        if i !=0 : 
            answer+=1

    return answer