[BOJ] 백준 11000번 강의실 배정 - 파이썬(Python)

문제

수강신청의 마스터 김종혜 선생님에게 새로운 과제가 주어졌다.

김종혜 선생님한테는 Si에 시작해서 Ti에 끝나는 N개의 수업이 주어지는데, 최소의 강의실을 사용해서 모든 수업을 가능하게 해야 한다. 

참고로, 수업이 끝난 직후에 다음 수업을 시작할 수 있다. (즉, Ti ≤ Sj 일 경우 i 수업과 j 수업은 같이 들을 수 있다.)

수강신청 대충한 게 찔리면, 선생님을 도와드리자!

입력

첫 번째 줄에 N이 주어진다. (1 ≤ N ≤ 200,000)

이후 N개의 줄에 Si, Ti가 주어진다. (0 ≤ Si < Ti ≤ 109)

출력

강의실의 개수를 출력하라.

 

https://www.acmicpc.net/problem/11000

11000번: 강의실 배정

 

💡 풀이 및 코드

import heapq
import sys
input = sys.stdin.readline

n = int(input())
array = []
for _ in range(n):
    array.append(tuple(map(int, input().split())))
array.sort(key=lambda x: x[0])

pq = []
heapq.heappush(pq, array[0][1])
for i in range(1, n):
    s = heapq.heappop(pq)
    if s <= array[i][0]:
        heapq.heappush(pq, array[i][1])
    else:
        heapq.heappush(pq, s)
        heapq.heappush(pq, array[i][1])

print(len(pq))

이전에 우선순위 큐를 사용할줄 몰라 미해결로 남겨두었던 문제이다.

시간 제한을 맞추기 위해서는 우선순위 큐를 사용해야 한다. 각각의 수업에 대하여 시작 시간을 기준으로 정렬해 놓은 후 종료 시간을 우선순위 큐인 pq에 푸시한다. 이때, 먼저 heappop을 통해 이전 수업들 중 가장 작은 종료 시간을 구한 후 현재의 수업 시작 시간이 해당 종료 시간보다 크거나 같으면 같은 강의실에 배정할 수 있으므로 현재의 수업 종료 시간을 pq에 대신 푸시한다. 그렇지 않다면 새로운 강의실이 필요하다는 의미이므로 pop으로 빼놨던 이전 수업의 종료 시간과 현재 수업의 종료시간을 함께 pq에 푸시한다. 최종적으로 pq의 길이가 강의실의 수가 된다.