[Python] 백준 11000 강의실 배정 (Heapq)

📌 문제

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

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

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

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

입력

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

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

출력

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

예제 입력 1

3
1 3
2 4
3 5

예제 출력 1

2

---

📌 풀이

💬 Code

import sys, heapq
input = sys.stdin.readline

n = int(input())
lesson = [list(map(int, input().split())) for i in range(n)]
lesson.sort()

room = []
heapq.heappush(room, lesson[0][1])

for i in range(1, len(lesson)):
    if room[0] <= lesson[i][0]:
        heapq.heapreplace(room, lesson[i][1])
    else:
        heapq.heappush(room, lesson[i][1])

print(len(room))

💡 Solution

• lesson: 강의시간을 담을 리스트
• room: 강의실 사용이 끝나는 시간을 담을 리스트 → 힙으로 만들어서 빨리 끝나는 것을 먼저 처리할 수 있도록 구현!

for i in range(1, len(lesson)):
    if room[0] <= lesson[i][0]:
        heapq.heapreplace(room, lesson[i][1])
    else:
        heapq.heappush(room, lesson[i][1])

• if room[0] <= lesson[i][0]
  가장 빨리 끝나는 강의의 종료시간 room[0]과 현재 강의의 시작시간 lesson[i][0]을 비교
  종료시간이 더 작거나 같다면 이 강의실을 그대로 쓸 수 있는 것
  따라서 진행중이었던 강의를 pop하고 현재 강의를 push

• else
  종료시간이 더 크다면 다른 강의실을 추가로 써야 하는 것
  따라서 현재 강의를 push

print(len(room))

• 강의실 개수를 출력