[Beakjoon] 백준 11000번 Python

Beakjoon 11000, 강의실 배정
solved.ac Gold5
Greedy 알고리즘, 우선순위 큐

문제

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

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

참고로, 수업이 끝난 직후에 다음 수업을 시작할 수 있다.

(즉, Ti ≤ Sj 일 경우 i 수업과 j 수업은 같이 들을 수 있다.)수강신청 대충한 게 찔리면, 선생님을 도와드리자!

입력

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

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

출력

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

Solution

import heapq

n = int(input())
data = []

for i in range(n):
    start, end = map(int, input().split())
    data.append([start, end])
data.sort()

classRoom = []
heapq.heappush(classRoom, data[0][1])

for i in range(1, n):
    if data[i][0] < classRoom[0]:
        heapq.heappush(classRoom, data[i][1])
    else:
        heapq.heappop(classRoom)
        heapq.heappush(classRoom, data[i][1])
print(len(classRoom))​

풀이

i번째 수업의 시작시간에 다른 수업으로 강의실이 사용 중이면 classRoom에 강의실을 추가합니다

강의실 사용의 끝나는 시간을 classRoom 리스트에 담고 강의실을 재사용할 수 있는 경우 앞에서 사용한 강의실의 끝나는 시간을 classRoom에서 pop 하고 재사용한 강의실의 끝나는 시간을 다시 classRoom리스트에 push 합니다.

classRoom은 오름차순으로 정렬됩니다.

• 총수업의 수를 입력하고 각 수업의 시작시간과 끝 시간을 입력합니다.
• 시작시간을 기준으로 해서 오름차순으로 data를 정렬합니다.
• 우선순위 큐에 강의실의 첫 번째 강의실의 사용이 끝나는 시간을 넣습니다.
• 수업이 시작하는 시간이 이전 수업의 끝나는 시간보다 먼저이면 강의실을 하나 추가합니다.
• 그렇지 않다면 강의실을 재사용할 수 있기 때문에 원래의 강의실 사용 시간을 제거하고 새로운 수업의 끝나는 시간을 classRoom에 넣습니다.

---

틀린 게 있다면 댓글 달아주세요~~

11000번: 강의실 배정