[py] - 프로그래머스 체육복 (Greedy)

문제

문제링크

풀이

• 여별의 체육복이 있는 학생(reserve)도 도난을 당했을 수 있다.
  동일존재값 제거

set_reserve = set(reserve)-set(lost)
set_lost = set(lost) - set(reserve)

• 양 옆 학생에게 빌려줄 수 있으므로 왼쪽 요소부터 탐색

코드

def solution(n, lost, reserve):
    set_reserve = set(reserve) - set(lost)
    set_lost = set(lost)-set(reserve)
    for i in set_reserve:
        if i-1 in set_lost:
            set_lost.remove(i-1)
        elif i+1 in set_lost:
            set_lost.remove(i+1)
    return n-len(set_lost)