강의실 배정

문제

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

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

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

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

입력

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

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

출력

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

해결방법

• 필요한 강의실의 최소값을 도출하기 위해 최소 힙을 사용한다.
• 먼저, 들어온 데이터 리스트를 정렬하고, 첫번째 원소의 끝나는 시간을 힙에 넣는다.
• 이후 모든 데이터로 for문을 돌리는데
  • for문 과정에서 다음 수업 시작 시간이 현재 힙의 최소값보다 작으면 강의실을 추가로 사용해야하기 때문에 힙에 넣어준다.
  • 다음 수업시작시간이 현재 가장 빨리 끝나는 수업시간보다 크거나 같으면 강의실을 이어서 사용할 수 있으므로, 힙에서 최소값을 제외하고, 해당 데이터의 끝나는 시간을 힙에 넣는다.
• 힙리스트의 길이를 출력한다.

import heapq 

N = int(input())

work = [list(map(int,input().split())) for _ in range(N)]

work.sort(key=lambda x:(x[0],x[1]))
heap = []
heapq.heappush(heap, work[0][1])

for i in range(1,N) :
    # 다음 수업 시작 시간이 현재 강의실 수업 시작의 최소값보다 작으면 
    #강의실 추가 사용 
    if work[i][0] < heap[0] :
        heapq.heappush(heap,work[i][1])
    #다음 수업시작시간이 현재 강의실 수업시작의 최소값보다 크거나 같으면 강의실 이어사 사용 
    else :
        heapq.heappop(heap)
        heapq.heappush(heap,work[i][1])
print(len(heap))