[백준] 11000: 강의실 배정 python 구현 / Priority Queue

문제

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

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

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

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

풀이

우선순위큐 알고리즘을 사용하여 1. 현재 수업 끝 시간 2. 다음 수업 시작 시간, 두 가지를 비교하여 구할 수 있다.

• 우선순위 큐를 쓰는 이유는 최소 힙을 사용하여 비교하게 될 현재 수업 끝 시간의 리스트를 계속 sorting해줄 수 있다. (아마도 내가 이해하기로는..)

1. 리스트로 받은 수업 시간들을 정렬 (시작 시간 기준)
2. 첫 번째 수업의 끝시간을 넣는다
3. for문을 통해 다음 수업의 시작시간과 비교하며
   1. 다음 수업 시작 시간 < 현재 수업 끝 시간 일 경우, 강의실이 추가적으로 필요하다 -> heappush
   2. 그게 아닐 경우, 현재 강의 실에서 계속 수업이 가능하다. 이후 비교가 되는 현재 끝 시간을 바꾸기 위해 -> heappop
4. heap에 있는 강의 실의 개수를 확인하자.

CODE

import sys, heapq

n = int(input())
classroom = [list(map(int, sys.stdin.readline().split())) for i in range(n)]
classroom.sort(key=lambda x: (x[0], x[1]))

room = []
heapq.heappush(room, classroom[0][1]) # 끝나는 시간을 하나 넣음

for i in range(1, n):
    if classroom[i][0] < room[0]:  # 다음 수업 시작시간 <<  현재 수업 끝나는 시간 : 강의실 + 1
        heapq.heappush(room, classroom[i][1]) # 다음 수업의 끝나는 시간을 하나 또 추가 ~
    else:   # 현재 강의실에서 계속 수업 가능
        heapq.heappop(room) # 비교시간을 바꾸기 위해 첫 번째꺼 pop
        heapq.heappush(room, classroom[i][1])

print(len(room))

코테 넘 어렵다 흑흑