탐욕법(Greedy) 체육복

탐욕법(Greedy) 체육복

 

 

def solution(n, lost, reserve):

 # reserve와 lost에 동시에 있는 학생은 먼저 제거 (자기 자신을 빌려 입기)
    lost_set = set(lost) - set(reserve)
    reserve_set = set(reserve) - set(lost)
    
    # 기본적으로 체육복을 가진 학생 수
    answer = n - len(lost_set)
    
    for i in reserve_set:
        if (i - 1) in lost_set :
            answer += 1
            lost_set.remove(i - 1)
        elif (i + 1) in lost_set :
            answer += 1
            lost_set.remove(i + 1)
        
    return answer;