백준(BOJ) 11000번 강의실배정 [그리디(Greedy)/알고리즘/파이썬(python)/큐]

11000번: 강의실 배정

 

 

문제

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

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

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

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

 

 

입력

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

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

 

출력

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

 

---

전에 풀어봤던 그리디 알고리즘이랑 달라서 계속 통과 못하고 있다가 내가 문제를 잘못이해했다는걸 깨달았다...

그 전에 풀었던 문제는 최대한 많은 수업을 들을 수 있는 방법 or 한 강의실에서 최대의 강의수 이런걸로 풀었어서 끝나는 시간이 가장 빠른 순서대로 정렬해서 다음 시작시간과 비교하는 방법이었는데, 이건 그 반대였던...😅 한참을 헤맸다!!

 

---

✍ 문제 이해

• 모든 수업을 들어야한다.
• 모든 수업을 듣기 위해서 필요한 가장 최소의 강의실 수는?

예제를 그림으로 나타낸다면,
같은 색으로 된 강의가 같은 강의실을 쓴다는 것이다.
그래서 총 2개의 강의실이 필요하다.

 

✍ 문제 풀이

강의 시작 리스트와 종료 리스트를 2차원 리스트로 묶고, 시작 시간을 기준으로 오름차순으로 정렬한다.

heapq를 사용하면 리스트가 항상 오름차순으로 정렬이 된다.

강의는 모두 들어야 하기 때문에 처음 시작했던 강의의 끝나는 시간을 heapq에 넣고, q리스트의 최소값과 다음 수업의 시작시간을 비교한다.

만약 다음 강의 시작 시간이 리스트의 최소값보다 큰 값이라면 최소값을 삭제한다.

다음 강의의 끝나는 시간을 q리스트에 넣는다.

마지막으로 q리스트의 길이를 출력하면 총 강의실 수가 나온다!

 

📝 코드

import heapq
import sys

def greedy_lecture_num(list):
    q = []
    heapq.heappush(q,list[0][1])
    
    for i in range(1, len(list)):
        if q[0] <= list[i][0]:
            heapq.heappop(q)
        heapq.heappush(q, list[i][1])
    return len(q)

N = int(sys.stdin.readline())
list = [[0 for j in range(2)] for i in range(N)]

for i in range(N):
    start, finish = map(int, sys.stdin.readline().split())
    list[i][0], list[i][1] = start, finish

list.sort(key=lambda x: x[0])

print(greedy_lecture_num(list))

 

---

python 코드로 제출할 때 input()함수 사용하면 안되겠다ㅠㅠ 자꾸 타임아웃이다..

앞으로 꼭 sys.stdin.readline() 사용해야지!!