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
|