[프로그래머스]체육복, python

프로그래머스 체육복 문제

 

def solution(n, lost, reserve):
    # remove common
    both = (set(lost) & set(reserve))
    lost = set(lost) - both
    reserve = set(reserve) - both
    
    count = n
    for sad_person in lost:
        if (sad_person+1) in reserve:
            reserve.remove(sad_person+1)
        elif (sad_person-1) in reserve:
            reserve.remove(sad_person-1)
        else:
            count -=1
            
    return count

set으로 lost와 reserve에 있는 공통된 사람의 제거한 뒤에,

lost를 반복문 돌리면서, 빌려줄 수 있는 사람이 있는지 체크하였다.

빌려줄 수 있는 사람이 있다면, 그 사람을 reserve배열에서 제거하고 

빌려줄 수 있는 사람이 없다면, count에서 1 을 뺐다.