본문 바로가기
🤔 알고리즘/시즌2-그리디-21.10.18~10.31

[백준] # 11000. 강의실 배정 (파이썬)

by 말랑한곰탱이 2021. 11. 22.

문제 링크

 

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

 

11000번: 강의실 배정

첫 번째 줄에 N이 주어진다. (1 ≤ N ≤ 200,000) 이후 N개의 줄에 Si, Ti가 주어진다. (0 ≤ Si < Ti ≤ 109)

www.acmicpc.net

 

문제 내용

 

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

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

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

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

 

입력

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

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

 

출력

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

 

IDEA

 

이전 수업의 종료시간이 현재 수업의 시작시간보다 작거나 같을 경우, 수업을 이어서한다. 클 경우, 새로운 강의실을 추가한다.

 

 

CODE

# 11000. 강의실 배정 - 골드5
# 시간초과... => input = sys.stdin.readline 했더니 성공!
impoty sys
import heapq
input = sys.stdin.readline
n = int(input())
time = []
for i in range(n):
    s, t = map(int,input().split())
    time.append((s,t))
time.sort()
cls = []
heapq.heappush(cls,time[0][1])
for i in range(1,n):
    # 이전 수업의 종료시간이 현재 수업의 시작시간보다 작거나 같을 경우, 수업을 이어서 한다.
    # cls[0]을 pop하고 현재 수업의 종료시간을 push.
    if cls[0] <= time[i][0]:
        heapq.heapreplace(cls,time[i][1])    # heapreplace : heap에서 가장 작은 항목을 팝하고 반환하며, 새로운 item도 푸시한다.
    # 이전 수업의 종료시간이 현재 수업의 시작시간보다 클 경우, 새로운 강의실 추가
    # 현재 수업의 종료시간 push
    else:
        heapq.heappush(cls,time[i][1])
print(len(cls))

 

사용 알고리즘

 

그리디

 

풀이 평

 

생각하기가 어렵고 재밌었다.