04_체육복

JIYOUNG KIM·2021년 12월 26일
0

프로그래머스

목록 보기
4/20
post-thumbnail

체육복을 잃어버린 사람이 다른 사람에게 빌려서 수업을 최대한 많이 들을 수 있도록 하는 것 하지만 빌릴 수 있는 체육복은 자신의 번호 +-1 이다.

  • 잃어버린 사람이 체육복을 빌리면 다른 사람에게 빌려줄 수 없으니 이를 set 함수를 통해서 제거해 줘야한다는 점
def solution(n, lost, reserve):

    holes = set(lost) - set(reserve)

    spares = set(reserve) - set(lost)

    answer = n - len(holes)

    for x in holes:

        if x - 1 in spares:

            spares.remove(x - 1)

            answer += 1

        elif x + 1 in spares:

            spares.remove(x + 1)

            answer += 1

    return answer
profile
데이터분석가

0개의 댓글