백준 1931번 회의실 배정 [Python]

[Silver I] 회의실 배정 - 1931

문제 링크

성능 요약

메모리: 51900 KB, 시간: 240 ms

분류

그리디 알고리즘, 정렬

제출 일자

2024년 4월 4일 23:15:07

문제 설명

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

입력

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

출력

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

풀이

# 처음에 시도한 틀린 풀이

import sys
from collections import deque
input = sys.stdin.readline
table = deque()
res = True
N = int(input())
for i in range(N):
    start, end = map(int, input().split())
    if not table:
        table.append((start, end))     
    else:
        for i,j in table:
            res = True
            if (i <= start and j <= start) or (i >= end and j >= end):
                continue
            else:
                res = False
                break
        if res:
            table.append((start, end))
sys.stdout.write("%d\n"%len(table))

테스트 케이스로 실행했을 때는 잘 실행되는 것처럼 보였다. 하지만 틀렸다는 판단을 받고 무엇이 틀렸는지 분석한 결과, 입력 순서가 개입한 것과 겹치는 시간을 판단하는 조건문이 문제였다는 것을 알았다.

일단 순서에 상관없이 최대 회의 수를 도출하기 위해 table리스트에 시간들을 다 담는다. 그 후에 가장 빨리 끝나는 회의 순으로 정렬 시킨다. 그래야 후에 다른 회의의 시작시간과 비교함으로써 겹치지 않게 시간순서를 지키는 최대값을 찾을 수 있다.

전체 코드

import sys

input = sys.stdin.readline
table = [] # 회의 시간을 담는 시간표

N = int(input())

for i in range(N):
    start, end = map(int, input().split())
    table.append((start, end))

# 회의가 빨리 끝나는 시간 순으로 정렬
table.sort(key=lambda x : (x[1], x[0]))        

endtime = table[0][1] # 초기 비교 시간 지정
cnt = 1
for i in range(1, N):

    # 각 회의의 끝 시간과 시작시간을 비교
    if table[i][0] >= endtime: # 시간이 겹치지 않으면
        endtime = table[i][1] # 다음 비교 대상
        cnt += 1
sys.stdout.write("%d\n"%cnt)