프로그래머스 체육복(파이썬)

이 문제의 경우 풀이는 매우 간단하나 딱 풀이를 떠올려서 구현하기가 너무 힘들다.

그리고 왼쪽에 있는 학생한테 먼저 빌려줘야 하는것도 키포인트

def solution(n, lost, reserve):
    
    a= set(lost)-set(reserve)    
    b= set(reserve)-set(lost)   #lost와 reserve 중복제거
    for i in b:     #체육복을 빌려줄 수 있을 때
        if i-1 in a:    #그 전 번호부터 확인해 봐야한다. 빌려줄 수 있을 때.
            a.remove(i-1)
        elif i+1 in a:  #그 다음 번호한테 빌려줄 수 있을 때
            a.remove(i+1)
            
                
    return n-len(a)