[Python/Programmers] 호텔 대실

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

프로그래머스

문제 설명

호텔을 운영 중인 코니는 최소한의 객실만을 사용하여 예약 손님들을 받으려고 합니다. 한 번 사용한 객실은 퇴실 시간을 기준으로 10분간 청소를 하고 다음 손님들이 사용할 수 있습니다.
예약 시각이 문자열 형태로 담긴 2차원 배열 book_time이 매개변수로 주어질 때, 코니에게 필요한 최소 객실의 수를 return 하는 solution 함수를 완성해주세요.

풀이

Code

import heapq
def solution(book_time):
    # 1. 힙 리스트 생성
    heap = []
    # 2. book_time 정렬
    # 3.
    for start, end in sorted(book_time) :
        start = list(map(int, start.split(':')))
        end = list(map(int, end.split(':')))
        # 3-1. 힙 리스트에 값이 있으면서 대실 종료 시간 + 청소 시간의 최솟값이 다음 대실 시작 시간보다 작거나 같을 경우
        if heap and (start[0] > heap[0][0] or (start[0] == heap[0][0] and start[1] >= heap[0][1])) :
            # 힙에서 제거
            heapq.heappop(heap)
        # 3-2. 다음 대실 종료 시간 + 청소 시간 삽입
        heapq.heappush(heap,(end[0], end[1] + 10)) if end[1] + 10 < 60 else heapq.heappush(heap,(end[0]+1, (end[1]+10)%60))
    # 4. 힙 리스트 길이 출력
    return len(heap)