[Python] 백준 11000번 - 강의실 배정 (골드 5)

혼자 힘으로 풀었는가?  X

알고리즘 분류
- 자료 구조
- 그리디 알고리즘
- 정렬
- 우선순위 큐

 

https://www.acmicpc.net/problem/11000

11000번: 강의실 배정

 

문제

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

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

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

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

입력

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

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

출력

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

 

---

 

정렬 후 앞의 끝나는 시간과 뒤의 시작시간을 비교하는 문제였다.

 

처음엔 단순히 반복문으로 처리했는데 시간 초과가 발생했다.

 

그래서 우선순위 큐로 문제를 푼다는데 난 이해가 잘 가지 않는다.

 

import sys
input = sys.stdin.readline

n = int(input())
data = []
visited = [0] * n
cnt = 0

for _ in range(n):
    s, t = map(int, input().split())
    data.append([s, t])
    
#data.sort(key = lambda x : x[1])
data.sort(key = lambda x : x[0])

import heapq
hq = []
heapq.heappush(hq, data[0][1])
cnt = 1 # 큐에 넣으면서 강의실 개설 됨
for i in range(1, n):
    if data[i][0] < hq[0]: # 다음 시작시간이 지금 끝나는 시간보다 빠르면
        heapq.heappush(hq, data[i][1]) # 새로운 강의실 개설
        cnt += 1
    else: #다음 시작 시간이 지금 끝나는 시간보다 크거나 같다면
        heapq.heappop(hq) # 새로운 강의로 시간 변경 위해 pop후 새 시간 push
        heapq.heappush(hq, data[i][1])
 
#print(len(hq))
print(cnt)

 

위의 과정들이 강의실을 배정하고 만들기 위한 작업이라는 것은 알겠는데

 

마지막에 큐의 길이가 생성된 강의실 개수라는점은 잘 이해가 되지 않는다.

 

그래서 나는 cnt 값을 추가하여 했는데 하고보니

 

남은 큐의 길이가 결국 pop되지 않은 값들이기 때문에 동일한 결과라고 볼 수 있다.