[백준]11000 강의실 배정 - 파이썬

문제

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

11000번: 강의실 배정

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

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

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

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

입력

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

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

출력

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

 

 

---

풀이

 

최소힙에 강의가 끝나는 시간을 저장해둔다.

새로운 강의의 시작시간 전에 이전 강의가 끝났다면 heap에서 끝난 강의를 pop하고 새로운 강의의 끝나는 시간을 저장한다.

만약 이전 강의가 끝나지 않았다면 pop없이 heap에 새로운 강의의 끝 시간을 추가해준다.

 

heap의 사이즈가 사용 중인 강의실의 개수이므로 최대 heap 사이즈를 저장해뒀다가 출력한다.

import sys
import heapq
input = sys.stdin.readline

n = int(input())
lect = []
for _ in range(n):
    lect.append(tuple(map(int, input().split())))
lect.sort()

maxroom = 0
heap = [0]

for l in lect:
    if heap[0] <= l[0]:
        heapq.heappop(heap)
    heapq.heappush(heap, l[1])
    maxroom = max(maxroom, len(heap))

print(maxroom)