[백준/BOJ] gold5 - 11000번 강의실 배정 (Python)

▶11000 - 강의실 배정

▶문제

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

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

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

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

 

▶입력

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

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

 

▶출력

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

728x90

▶풀이

heapq를 만들어서 문제를 풀었다.

그때그때 강의실 중 가장 빨리 끝나는 강의실을 찾아 그 강의실이 끝나는 시간과 새로운 강의가 시작하는 시간을 비교했다.

들어갈 수 있으면 끝나는 시간을 업데이트해주고, 들어갈 수 없다면 새로 만들어서 넣어줬다.

 

단 이문제는 python3는 시간 초과가 났고, pypy3는 제대로 풀렸다.

import heapq


n = int(input())
times = []
for i in range(n):
    s, t = map(int, input().split())
    times.append([s, t])
times.sort()

room = []
heapq.heapify(room)
heapq.heappush(room, times[0][1])
for i in range(1, n):
    if room[0] <= times[i][0]:
        heapq.heappop(room)
        heapq.heappush(room, times[i][1])
    else:
        heapq.heappush(room, times[i][1])
print(len(room))