[코딩테스트연습] Programmers > 코딩테스트 연습 > 연습문제 > 호텔 대실 - python

[Programmers] 호텔 대실 (Python) - 테크가이던스 - 티스토리

문제 설명

호텔을 운영 중인 코니는 최소한의 객실만을 사용하여 예약 손님들을 받으려고 합니다. 한 번 사용한 객실은 퇴실 시간을 기준으로 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" 까지로 주어집니다.
    • 예약 시각이 자정을 넘어가는 경우는 없습니다.
    • 시작 시각은 항상 종료 시각보다 빠릅니다.

---

입출력 예

book_timeresult

[["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개의 방이 필요합니다.

 

💡 문제 풀이

보통 코딩테스트의 시간이 포함된 문제의 경우 시간을 모두 숫자로 바꾸어서 해결하는게 일반적이라고 생각하면 좋습니다.

그리고 본 문제는 heap을 사용하였습니다.

 

Heap이란

- complete binary tree이면서

- heap property를 만족해야 합니다.

   ㄴ max heap property : 부모는 자식보다 크거나 같다 (일반적으로 max heap을 다룸)

   ㄴ min heap property : 부모는 자식보다 작거나 같다

 

- Heap은 일차원 배열로 표현 가능합니다.

- 파이썬에서는 힙 큐 알고리즘을 라이브러리로 제공하며 아래 url 참고 부탁드립니다.

- python heapq : # https://docs.python.org/ko/3/library/heapq.html

heapq — Heap queue algorithm

heapq.heappush(heap, item)

힙 불변성을 유지하면서, item 값을 heap으로 푸시합니다.

 

heapq.heappop(heap)

힙 불변성을 유지하면서, heap에서 가장 작은 항목을 팝하고 반환합니다. 힙이 비어 있으면, IndexError가 발생합니다. 팝 하지 않고 가장 작은 항목에 액세스하려면, heap[0]을 사용하십시오.

 

heapq.heappushpop(heap, item)

힙에 item을 푸시한 다음, heap에서 가장 작은 항목을 팝하고 반환합니다. 결합한 액션은 heappush()한 다음 heappop()을 별도로 호출하는 것보다 더 효율적으로 실행합니다.

 

#Python 코드

from heapq import heappop, heappush

def solution(book_time):
    answer = 1
    new_book_time = []
    heap = []

    for b in book_time:
        check_in = int(b[0][:2]) * 60 + int(b[0][3:5])
        check_out = int(b[1][:2]) * 60 + int(b[1][3:5]) + 10 #룸 클리닝 10분 추가
        new_book_time.append([check_in,check_out])
    new_book_time.sort(key = lambda x:(x[0],x[1]))
    
    print(new_book_time)
    
    for s, e in new_book_time:
        if not heap:
            heappush(heap,e)
            continue
        if heap[0] <= s:
            heappop(heap)
        else:
            answer += 1
        heappush(heap,e)
        
    return answer

 

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges