(Python/LV1)체육복

문제출처: https://school.programmers.co.kr/learn/courses/30/lessons/42862

프로그래머스

 

풀이:

def solution(n, lost, reserve):

    reserve_p = set(reserve) - set(lost) # 여유있는사람
    lost_p = set(lost) - set(reserve)	# 잃어버린사람


    for i in reserve_p:
        be = i - 1
        af = i + 1
        if be in lost_p:		# 전 숫자가 있는경우
            lost_p.remove(be)		# 전 숫자를 잃어버린 사람에서 삭제
        elif af in lost_p:		# 반대경우
            lost_p.remove(af)
    return n - len(lost_p)		# 전체인원 - 옷 못 빌린사람 

print(solution(5,[2, 4],[3]))