[Programmers] 호텔 대실 바로가기
호텔을 운영 중인 코니는 최소한의 객실만을 사용하여 예약 손님들을 받으려고 합니다. 한 번 사용한 객실은 퇴실 시간을 기준으로 10분간 청소를 하고 다음 손님들이 사용할 수 있습니다.
예약 시각이 문자열 형태로 담긴 2차원 배열 book_time
이 매개변수로 주어질 때, 코니에게 필요한 최소 객실의 수를 return 하는 solution 함수를 완성해주세요.
book_time | result |
---|---|
[["15:00", "17:00"], ["16:40", "18:20"], ["14:20", "15:20"], ["14:10", "19:20"], ["18:20", "21:20"]] | 3 |
[["09:10", "10:10"], ["10:20", "12:20"]] | 1 |
[["10:20", "12:30"], ["10:20", "12:30"], ["10:20", "12:30"]] | 3 |
입출력 예 #1
위 사진과 같습니다.
입출력 예 #2
첫 번째 손님이 10시 10분에 퇴실 후 10분간 청소한 뒤 두 번째 손님이 10시 20분에 입실하여 사용할 수 있으므로 방은 1개만 필요합니다.
입출력 예 #3
세 손님 모두 동일한 시간대를 예약했기 때문에 3개의 방이 필요합니다.
객실은 손님에게 할당한다. 할당하는 방법은 다음과 같다.
book_time_ref = [(int(s[:2]) * 60 + int(s[3:]), int(e[:2]) * 60 + int(e[3:])) for s, e in book_time]
book_time_ref.sort()
["14:10", "19:20"]
로 주어진다면 분단위의 숫자형으로 값을 변경해준다."14:10"
→ 14 * 60 + 10 = 850
"19:20"
→ 19 * 60 + 20 = 1160
heap = []
for s, e in book_time_ref:
if not heap:
heappush(heap,e)
continue
if heap[0] <= s:
heappop(heap)
else:
answer += 1
heappush(heap,e+10)
s
)을 기준으로 정렬된 예약 시간을 통해 객실을 할당한다.heap[0]
)이 현재 대실 시작 시각(s
)보다 같거나 빠르다면 해당 객실을 현재 손님에게 할당한다.heap[0]
)이 현재 대실 시작 시각(s
)보다 느리다면 새로운 객실(answer
)을 추가한다.from heapq import heappop, heappush
def solution(book_time):
answer = 1
# "HH:MM" → HH * 60 + MM
book_time_ref = [(int(s[:2]) * 60 + int(s[3:]), int(e[:2]) * 60 + int(e[3:])) for s, e in book_time]
book_time_ref.sort()
heap = []
for s, e in book_time_ref:
if not heap:
heappush(heap,e)
continue
if heap[0] <= s:
heappop(heap)
else:
answer += 1
heappush(heap,e+10)
return answer
안녕하세요. 해당 문제 풀이 과정을 찾다가 들렸습니다.
힙을 쓸 생각을 못 해서 정답은 맞췄지만 시간이 느렸는데 덕분에 더 효율적인 코드를 찾을 수 있었습니다. ㅎㅎ
그치만 코드에서 오류를 발견했습니다. heap 리스트가 비워있다면 e 값을 push 하는데 이 경우에도 e+10 을 push 하는 것이 맞는 것 같아 댓글 남깁니다!
좋은 풀이 감사합니다.