[알고리즘]기초2-체육복

체육복

https://programmers.co.kr/learn/courses/30/lessons/42862

def solution(n, lost, reserve):
    answer = 0
    
    stu=[1]*n
    
    for i in lost:
        stu[i-1]-=1
    
    for i in reserve:
        stu[i-1]+=1
        
    for i in range(n):       
        if stu[i]==0:
            if i!=0 and stu[i-1]>1:
                stu[i-1]-=1
                stu[i]+=1
            elif i!=n-1 and stu[i+1]>1:
                stu[i+1]-=1
                stu[i]+=1

    answer=n-stu.count(0)
    return answer

• [1]*n인 리스트 stu를 선언
• lost 원소에 해당하는 index의 값을 -1
• reserve의 원소에 해당하는 index의 값을 +1
• for문을 이용하여 stu의 값이 0인 원소의 전, 후를 확인하여 빌린 학생은 +1, 빌려준 학생은 -1
• count(0)을 하여 체육복이 없는 학생의 수를 세서 전체 학생 수에서 뺀 값을 answer로 반환