[프로그래머스/Python] Lv 1. 체육복

⭐ 문제 링크

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

프로그래머스

 

⭐ 풀이 코드

def solution(n, lost, reserve):
    answer = 0
    
    # 여벌옷 없는 lost
    real_lost = set(lost) - set(reserve)
    
    # 도난당하지 않은 reserve
    real_reserve = set(reserve) - set(lost)
    
    for r in real_reserve:
        if r-1 in real_lost:  # 앞
            real_lost.remove(r-1)
        elif r+1 in real_lost:  # 뒤
            real_lost.remove(r+1)
    
    answer = n - len(real_lost)
    
    return answer