[BOJ] 11000 강의실 배정 (Python3)

문제

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

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

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

입력

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

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

출력

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

예제

예제 입력 1  
3
1 3
2 4
3 5

예제 출력 1  
2

문제풀이

맨 처음 수업의 끝나는시간을 큐에 넣어주고, 그 다음 수업부터의 시작시간을 queue[0] 과 비교한다. 만약 queue[0] 보다 시작시간이 빠르다면, 그 강의실은 아직 수업중이므로 새로운 강의실을 배정해주어야 한다. 따라서 heappush로 강의실을 하나 더 배정해준다. 만약 queue[0] 보다 시작시간이 같거나 느리다면, 그 강의실을 사용할 수 있는 것이므로 기존의 끝나는시간을 heappop해주고 새로운 끝나는시간을 push해주면 된다.

코드

import heapq

n = int(input())

time = []

for i in range(n):
    time.append(list(map(int, input().split())))

time.sort(key = lambda x : x[0])

queue = []
heapq.heappush(queue,time[0][1])

for i in range(1,len(time)):
    if queue[0] > time[i][0]:
        heapq.heappush(queue, time[i][1])
    else:
        heapq.heappop(queue)
        heapq.heappush(queue, time[i][1])
        
print(len(queue))

3
1 3
2 4
3 5
2