[백준 / 그리디] 1931 : 회의실 배정 (python)

문제 설명

문제

한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.

입력

첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.

예제 입력 1

11
1 4
3 5
0 6
5 7
3 8
5 9
6 10
8 11
8 12
2 13
12 14

예제 출력 1

4

힌트

(1,4), (5,7), (8,11), (12,14) 를 이용할 수 있다.

추가 예제

4
0 0
0 0
0 0
0 0

# 결과 4

3
5 5
3 5
2 5

# 결과 2

 

풀이

from queue import PriorityQueue
import sys

input = sys.stdin.readline

N = int(input())
q = PriorityQueue()

for i in range(N):
    data = list(map(int, input().split()))
    data.sort(reverse=True)
    q.put(data)

answer = 0
before = 0

while q.qsize() > 0:
    cur = q.get()

    if before == 0:
        before = cur
        answer += 1
    elif before[0] <= cur[1]:
        answer += 1
        before = cur

print(answer)

• 우선순위큐(PriorityQueue)를 이용하여 풀이
• 끝나는 시간을 기준으로 우선순위 큐에 저장함. 이를 위해 내림차순 정렬을 한 뒤 큐에 저장함.
  끝나는 시간이 짧은 순으로 큐에 저장되는데 이는 회의 시간이 짧고, 시작 시간이 빠름을 의미함.
• 현재 회의의 끝나는 시간보다 다음 회의의 시작 시간이 같거나 클 때 answer에 추가.

처음에는 현재 회의의 끝나는 시간과 다음 회의의 시작 시간이 겹치면 안된다고 생각해서 계속 틀림..