[Python][누적합] 호텔 대실 - 프로그래머스 Lv.2

[문제]

문제 설명

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

 

입출력 예

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

 

[알고리즘]

• 누적합 문제이다. 

1. 분 단위로 배열을 만든 다음, 시작 타임에 1을 더해주고, 끝 타임+1에 1을 빼준다. 
아래 예시는 시작 3분, 끝 7분 / 시작 4분, 끝 8분 일 경우이다. 

이미지 출처 : https://ksb-dev.tistory.com/269

 

2. 이 배열을 1번 반복하여, 이전 배열값을 현재의 배열 값에 더하여 업데이트한다. 
이 배열의 값들이 해당 시간에 겹치는 스케쥴의 수이다. 

이미지 출처 : https://ksb-dev.tistory.com/269

 3. 이 배열의 max 값을 출력하면 끝 

 

[코드]

def solution(book_time):
    answer = 0
    times = [0] * (25*60)
    
    for i, (start, end) in enumerate(book_time):
        h1, m1 = map(int, start.split(':'))
        h2, m2 = map(int, end.split(':'))
        
        times[h1*60 + m1] += 1
        times[h2*60 + m2+10] -= 1
    
    for i in range(1, len(times)):
        times[i] += times[i-1]
    
    return max(times)

 

[누적합을 적용하기 전 틀린 코드]

def solution(book_time):
    answer = 0
    rooms = []
    
    def compare_t(t1, t2):
        h1, m1 = map(int, t1.split(':'))
        h2, m2 = map(int, t2.split(':'))
        
        if h1 > h2 : 
            return False
        elif h1 < h2 :
            if abs(h1 - h2) == 1 : ## 13시 55분, 14시 00분일 경우를 위함
                if m1 >= 50 and m2 <=10 : 
                    if 60-(m1-m2) < 10 : 
                        return False
                    else :
                        return True
                else : 
                    return True
            else : 
                return True 
        else : 
            if m1 + 10 <= m2  : 
                return True 
            else :
                return False

    def check_room(start, end):
        for i, room in enumerate(rooms):
            is_possible = True
            for s, e in room:
                if compare_t(e, start) or compare_t(end, s):
                    pass
                else :
                    is_possible = False
                    break
            if is_possible : 
                rooms[i].append((start, end))
                return
        rooms.append([(start, end)])
        
    for i, (start, end) in enumerate(book_time) :
        check_room(start, end)
    print(rooms)
        
    return len(rooms)