[프로그래머스 Lv2] 호텔 대실: 특정 시간 소요 배치하기(누적 합, 힙)

프로그래머스 Lv2. 호텔 대실

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

제한사항
1 ≤ book_time의 길이 ≤ 1,000
book_time[i]는 ["HH:MM", "HH:MM"]의 형태로 이루어진 배열입니다
[대실 시작 시각, 대실 종료 시각] 형태입니다.
시각은 HH:MM 형태로 24시간 표기법을 따르며, "00:00" 부터 "23:59" 까지로 주어집니다.
예약 시각이 자정을 넘어가는 경우는 없습니다.
시작 시각은 항상 종료 시각보다 빠릅니다.

입출력 예

입출력 예 설명
입출력 예 #1
example1

위 사진과 같습니다.

입출력 예 #2
첫 번째 손님이 10시 10분에 퇴실 후 10분간 청소한 뒤 두 번째 손님이 10시 20분에 입실하여 사용할 수 있으므로 방은 1개만 필요합니다.

입출력 예 #3
세 손님 모두 동일한 시간대를 예약했기 때문에 3개의 방이 필요합니다.

 

---

[대표 풀이] : 누적합

- 위의 Ex1 그림 자체를 time_table로 형상화한 것. booked 된 부분을 time_table의 해당 시간대에 1을 더해 표현한다. 

- 모든 예약 건에 대해서 예약된 시간대를 1을 더하면, 최대 중복된 시간대의 값이 필요한 방의 개수가 된다는 아이디어. 

- 내 풀이보다 더 직관적이었던 점이 좋았다. 

def solution(book_time):
    time_table = [0 for _ in range(60 * 24)]
    for start, end in book_time:
        start_minutes = 60 * int(start[:2]) + int(start[3:])
        end_minutes = 60 * int(end[:2]) + int(end[3:]) + 10

        if end_minutes > 60 * 24 - 1:
            end_minutes = 60 * 24 - 1

        for i in range(start_minutes, end_minutes):
            time_table[i] += 1
    return max(time_table)

 

 

---

[내 풀이] : 힙 

- 특정 시간 소요 배치

- 가장 이른 값만 중요하므로 힙을 사용했다.

import heapq as hq
def solution(book_time):
    
    # book_time 값 정리
    book_time = [[int(x[0][:2])*60+int(x[0][3:5]), int(x[1][:2])*60+int(x[1][3:5])] for x in book_time]
    
    # 시작 시간 이른 순 정렬 먼저 
    book_time.sort()

    # 메인 루프
    room = []
    hq.heappush(room, 0)
    clean_time = 10
    for check_in, check_out in book_time:
        if room[0]+clean_time <= check_in:
            hq.heappop(room)
            hq.heappush(room,check_out)
        else:
            hq.heappush(room, check_out)
    return len(room)