프로그래머스 - 체육복

체육복을 도난당한 학생의 번호와 여벌이 있는 학생의 번호가 같을 경우를 고려해야하는 문제.

def solution(n, lost, reserve):

    # 체육복을 도난당한 학생이 여벌이 있을경우를 고려하여 둘 다 제거
    real_lost = list(set(lost) - set(reserve))
    real_reserve = list(set(reserve) -set(lost))

    for i in range(len(real_reserve)):
        if real_reserve[i] -1 in real_lost:
            real_lost.remove(real_reserve[i] -1)
        elif real_reserve[i] + 1 in real_lost:
            real_lost.remove(real_reserve[i] + 1)

    answer = n - len(real_lost)
    return answer

 

---

 

programmers.co.kr/learn/courses/30/lessons/42862

코딩테스트 연습 - 체육복