프로그래머스 1단계 "체육복"

문제

프로그래머스 체육복

풀이

그리디알고리즘을 이용하여 풀었다.

중복을 없애기위해 set을 사용하였고, 여벌체육복을 가져온 학생이 도난을 당했을 수 있으므로
도난자와 여벌 체육복 휴대자의 집합을 서로 빼주었다.

Python 코드

def solution(n, lost, reserve):
  lost1 = set(lost)- set(reserve)
  reserve1 = set(reserve) - set(lost)

  for a in reserve1:
    if a-1 in lost1:
      lost1.remove(a-1)
    elif a+1 in lost1:
      lost1.remove(a+1)

  return n-len(lost1)