BOJ: 11000

Posted on In

문제

수강신청의 마스터 김종혜 선생님에게 새로운 과제가 주어졌다.
김종혜 선생님한테는 Si에 시작해서 Ti에 끝나는 N개의 수업이 주어지는데, 최소의 강의실을 사용해서 모든 수업을 가능하게 해야 한다.
참고로, 수업이 끝난 직후에 다음 수업을 시작할 수 있다. (즉, Ti ≤ Sj 일 경우 i 수업과 j 수업은 같이 들을 수 있다.)
수강신청 대충한 게 찔리면, 선생님을 도와드리자!

입력

첫 번째 줄에 N이 주어진다. (1 ≤ N ≤ 200,000)
이후 N개의 줄에 Si, Ti가 주어진다. (0 ≤ Si < Ti ≤ 109)

출력

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

Solution

입력 구현

1
2
3
4
5
6
7
8
9
import sys
import heapq
read = sys.stdin.readline

N = int(read())
ST = []

for i in range(N):
  ST.append(list(map(int, read().split())))

가장 빨리 시작하는 수업에 대해 강의실 배정 후 가장 빨리 끝나는 강의실에 대해 다음 수업 진행

1
2
3
4
5
6
7
8
9
10
11
12
ST.sort(key = lambda x: (x[0], x[1]))
G = []
heapq.heappush(G, ST[0][1])

for i in range(1, N):
  if ST[i][0] < G[0]:
    heapq.heappush(G, ST[i][1])
  else:
    heapq.heappop(G)
    heapq.heappush(G, ST[i][1])

print(len(G))
  • 강의실을 나타내는 힙 G에 가장 빠르게 시작하는 수업을 배정하며 수업의 종료시간을 입력한다.
  • 시작시간을 기준으로 오름차순 정렬된 배열 ST에 대해 다음으로 빠르게 시작하는 수업의 시작 시간 ST[i][0]이 제일 빠르게 마치는 수업의 종료 시간 G[0]보다 빠를 경우 새로운 강의실을 heapq.heappush로 배정해준다.
  • 반대의 경우 heapq.heappop으로 강의실을 비워주고 heapq.heappush로 다음 수업을 해당 강의실로 배정해준다.
  • 최종적으로 강의실 G의 길이를 출력한다.