[프로그래머스] 체육복 - 파이썬

Programmers

체육복

https://school.programmers.co.kr/learn/courses/30/lessons/42862

프로그래머스

문제

코드

• 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있기 때문에 reserve에 있는 값이 lost에도 있을 수 있는 경우를 고려해야함.

def solution(n, lost, reserve):

    lost.sort()
    reserve.sort()

    for i in reserve[:]:
        if i in lost:
            reserve.remove(i)
            lost.remove(i)

    for i in reserve:

        if i-1 in lost:
            lost.remove(i-1)

        elif i+1 in lost:
            lost.remove(i+1)

    return n-len(lost)