[Algorithm] 프로그래머스 체육복 | 파이썬 (그리디)

※ 42862 체육복

https://school.programmers.co.kr/learn/courses/30/lessons/42862?language=python3

프로그래머스

문제 해결 TIP

우선, 제한 사항 중 여벌 체육복을 가져온 학생이 도난당한 경우를 고려한 경우를 정리해준다. 이후 나눠줄 수 있는 앞 뒤  번호와 도난당한 학생의 번호를 비교하여 한 명씩 제외한다. 마지막으로, 남아있는 도난당한 학생 수를 전체에서 빼준다. 앞 번호부터 순차적으로 나눠주는 것이 중요함으로 정렬을 이용한다.

 

전체 코드

def solution(n, lost, reserve):
    lost.sort()
    reserve.sort()
    _reserve = [r for r in reserve if r not in lost]
    _lost = [l for l in lost if l not in reserve]
    for r in _reserve:
        f = r - 1
        b = r + 1
        if f in _lost:
            _lost.remove(f)
        elif b in _lost:
            _lost.remove(b)
    return n - len(_lost)