강의실 배정 [Python]

문제

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

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

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

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

입력

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

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

출력

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

풀이

#우선순위 큐, 힙큐
import heapq

n = int(input())
class_list = []

for _ in range(n):
    class_list.append(tuple(map(int, input().split())))

class_list.sort(key=lambda x:x[0])
q = [class_list[0][1]]

for i in range(1, n):
    if q[0] > class_list[i][0]: #강의실을 쓰고 있으므로 하나 추가
        heapq.heappush(q, class_list[i][1])
    else: #다 쓴 강의실 이어 사용
        heapq.heappop(q)
        heapq.heappush(q, class_list[i][1])

print(len(q))

우선순위 큐를 이용하여 풀이하였다.
강의를 시작 시간대로 정렬하고, 힙큐에 저장된 강의 완료 시간과 새롭게 탐색하는 강의들의 시작 시간을 비교하며 풀이하였다.

힙큐에 있는 최소 힙의 강의 완료 시간보다 for문을 돌며 탐색하는 강의 시작 시간이 크다면 이어서 강의실을 쓰면 되므로, pop하고 push하여 강의실 정보를 갱신하였다.
그 반대라면 강의실을 새로 열어야 하므로 push 연산만 수행하도록 하였다.