[프로그래머스-Lv1] 체육복 / Python

코딩테스트 연습 - 체육복

import sys

input = sys.stdin.readline
n = int(input())
lost = list(map(int, input().split()))
reserve = list(map(int, input().split()))


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


print(solution(n, lost, reserve))

 

해설

여벌의 체육복을 가지고 있는 학생이 체육복을 잃어버릴 수 있는 경우를 위해 set을 통해 다시 재정의해줬다.

setReserve의 값들에서 -1, +1한 값이 setLost에 있으면 해당 값을 setLost에서 제거한 후, 마지막엔 총 인원-setLost의 길이를 리턴해준다.