체육복 - Python, set

 

 

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

프로그래머스

 

 

 

풀이 - 나의 생각

set의 차집합을 이용해서

 

체육복을 빌려 줄 수 있는 경우를 체크합니다.

 

이미 여분이 있는 경우에도 차집합을 통해 자동으로 걸러집니다.

 

코드

 

def solution(n, lost, reserve):
    real_lost = set(lost) - set(reserve)
    real_reserve = set(reserve) - set(lost)
    
    for no in sorted(real_reserve):
        if no - 1 in real_lost:
            real_lost.remove(no-1)
        elif no + 1 in real_lost:
            real_lost.remove(no+1)
    
    return n - len(real_lost)

 

 

 

 

 

 

 

다음에 또 봐요