백준 11000번 강의실 배정 파이썬 풀이 heap

강의실 배정 성공

 

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율

1 초

256 MB

36634

10772

7915

28.905%

문제

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

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

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

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

입력

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

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

출력

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

예제 입력 1 복사

3
1 3
2 4
3 5

예제 출력 1 복사

2

 

 

 

 

 

import heapq

import sys
input = sys.stdin.readline

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

for i in range (n):
    a, b=list(map(int, input().split()))
    arr.append([a,b])
    

arr.sort() #시작시간으로 정렬

heaps=[] #minheap생성!! 부모가 자식보다 작고, 자식간에는 대소x
heapq.heappush(heaps, arr[0][1]) # 첫 수업의 끝시간을 heap넣는다 

for i in range (1, n):
    if (heaps[0]> arr[i][0] ):
        heapq.heappush(heaps, arr[i][1]) #heap에 원소추가
    else: #이어서 가능하다면
        heapq.heappop(heaps) #가장 작은 원소(root노드) 삭제해서 갱신
        heapq.heappush(heaps, arr[i][1]) #heap에 원소추가

#ex [[1,3]] -> [[1, 3], [2, 4]] -> [[2, 4], [3,5]] 이렇게 들어가게된다
#*실제로는 끝시간만 들어가는데 보기편하라고 위와같이 표기함
                
print(len(heaps)) #heaps의 크기를 반환하면 됨

 

지난번 푼 회의실 문제랑 비슷해서 그렇게 풀다가 이중 for문이 되어서

heap으로 풀게 된 문제다.

최소힙트리로 부모는 자식보다 작고, 자식간에는 대소관계가 없다.