파이썬 그리디 알고리즘 백준 11000 강의실 배정

문제

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

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

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

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

입력

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

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

출력

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

풀이

최소한의 강의실을 사용하여 모든 수업을 해야 한다.

빠지는 수업이 있으면 안되므로 입력받은 수업을 시작하는 시간 기준으로 오름차순 정렬한다.

이 문제는 heapq를 이용하여 해결할 수 있는데 반복문을 순회하며 새로운 수업을 검사할 때

이전 수업 시간의 끝나는 시간 중 가장 빨리 끝나는 시간과 비교하면 되기 때문이다.

따라서 우선 제일 일찍 시작하는 수업의 끝나는 시간을 nheaps에 저장한다.

그리고 수업 리스트를 인덱스 1부터 시작하여 시작하는 시간과

nheaps에 있는 가장 빨리 끝나는 시간 (nheaps[0])을 비교하여 끝나는 시간이 새로운 수업의 시작하는 시간보다

같거나 작은 경우 기존의 끝나는 시간을 pop해서 제거하고 새 수업의 끝나는 시간을 push한다.

그렇지 않은 경우는 새로운 강의실을 써야 하므로 새로운 수업의 끝나는 시간을 그대로 push한다.

반복문이 끝나고 nheaps의 길이가 필요한 총 강의실 갯수 이다.

 

import sys
import heapq
n = int(sys.stdin.readline().rstrip("\n"))
classes = []
for i in range(n):
    classes.append(list(map(int, sys.stdin.readline().rstrip("\n").split())))
classes = sorted(classes, key=lambda x:x[0])
nheapps = []
heapq.heapify(nheapps)
heapq.heappush(nheapps, classes[0][1])

for i in range(1,n):
    if(nheapps[0] <= classes[i][0]):
        heapq.heappop(nheapps)
        heapq.heappush(nheapps,classes[i][1])
    else:
        heapq.heappush(nheapps,classes[i][1])

print(len(nheapps))