[백준 11000번] 강의실 배정 - 파이썬(python)

11000번: 강의실 배정

분류

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

문제

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

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

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

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

입력

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

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

출력

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

풀이

• 주어진 시간에 따라 강의실을 최소 몇개 사용해야 하는지 출력해야 한다.
• 시간을 수업이 시작한 순서대로 정렬한다.
• 수업이 끝나는 사간을 저장하고 다음 수업 시작 시간을 비교하여 작거나 같은 경우라면 pop하고 끝나는 시간으로 변경하여 rlt배열에 넣는다.
• 마지막에 rlt의 원소 개수가 답이다.

코드

import sys, heapq
input = sys.stdin.readline

q=[]
N = int(input())

for _ in range(N):
    a, b = map(int, input().split())
    q.append([a,b])
q.sort()
# 결과 배열
rlt = []
for a, b in q:
    # 배열이 비어있는 경우 끝나는 시간을 push
    if not rlt:
        heapq.heappush(rlt, b)
    else:
        # 비교하여 작거나 같다면 pop하고 push하고 아니라면 pop없이 push한다.
        if rlt[0] <= a:
            heapq.heappop(rlt)
        heapq.heappush(rlt, b)

print(len(rlt))