BAEKJOON #11000 (Greedy)

nathan·2021년 7월 2일

알고리즘문제

목록 보기

10/102

강의실 배정

시간 제한	메모리 제한
1초	256 MB

문제

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

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

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

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

입력

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

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

출력

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

입출력 예시

예제 입력 1

3
1 3
2 4
3 5

예제 출력 1

2

풀이

생각

처음에는 강의실 개수를 출력하는 것이 아니라 가장 많은 강의를 들으려면 어떻게 해야하는지로 착각하고 실수했었다. (비슷한 유형이라고 다 같은 문제는 아니다.)
파이썬 자료구조 중 heapq를 이용하였는데, 이를 이용하기 전에는 시간초과로 고생 좀 했다.
heapq가 자동으로 최소 값을 저장하는 구조이기 때문에 유용했다.

풀이 설명

우선 sorted를 통해 시작시간이 빠른 순서대로 정렬하였다.
최소 힙을 만든 후 가장 시작시간이 빠른 회의의 종료시간을 push 했다.
1부터 n-1까지 돌면서, 힙의 최솟값이 새로운 회의의 시작시간보다 작거나 같으면, 같은 회의실을 이용할 수 있다는 말이므로, 힙의 최솟값을 pop하고, 새로운 회의의 종료시간을 넣어주었다.
만약 힙의 최솟값이 새로운 회의의 시작시간보다 크다면, 같은 회의실을 이용할 수 없다는 소리이므로, pop 없이 새로운 회의의 종료시간을 push했다.
즉, 모든 회의를 다 탐색하는 시점에는 heap에 남아있는 원소의 개수가 이용한 회의장이 된다.

python code

# 백준 11000번
from sys import stdin
import heapq
# 최소의 강의실 사용하여 모든 수업을 가능케 해야함.. 
# 강의실 개수를 출력해야 함
def room(n, time):
    newList = sorted(time, key=lambda x: (x[0], x[1]))
    heap = []
    heapq.heapify(heap)
    heapq.heappush(heap, newList[0][1])   # 시작시간이 가장 빠른 회의를 추가함

    for i in range(1, n): # 만약 한 회의실에 회의를 더 배정할 수 있다면,(heap에 들어간 회의의 종료시간 <= 새로운 회의의 시작시간)
        if heap[0] <= newList[i][0]: 
            heapq.heappop(heap) # 해당 원소를 heap에서 제외시키고
            heapq.heappush(heap, newList[i][1])   # 힙에 새로 원소를 추가한다.
        else:   
            # heap에 들어간 회의의 종료시간이 새로운 회의의 시작시간을 넘기면,
            heapq.heappush(heap, newList[i][1])   # 힙에 새로 원소를 추가한다.

    return len(heap)

n = int(stdin.readline())
time = []
for i in range(n):
    time.append(tuple(map(int, stdin.readline().split())))

result = room(n, time)
print(result)