[1379] 강의실2 파이썬 python

이 문제는 예시랑 내가 짠 코드가 정확히 일치하지 않아서 로직이 잘못 된 줄 알고 계속 고민하느라 오랜 시간이 소요됐다.

문제

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

1379번: 강의실 2

코드

arr에는 시작 시간 기준으로 sorting을 하고, heapq에는 끝나는 시간 기준으로 우선순위 큐 정렬하기

from heapq import heappush, heappop

pq = []
n = int(input())
room = [0] * n
arr = []
for _ in range(n):
    idx, s, e = map(int, input().split())
    arr.append([s, e, idx - 1])

arr = sorted(arr, key=lambda x:(x[0]))
ans = 0
for s, e, idx in arr:
    if pq and pq[0][0] <= s:
        room[idx] = room[pq[0][2]]
        heappop(pq)
    else:
        ans += 1
        room[idx] = ans
        
    heappush(pq, [e, s, idx])

print(ans)
for i in room[1:]:
    print(i)