백준 1931 회의실 배정 - 파이썬

문제.

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

 

입력.

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

 

출력.

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

 

풀이.

처음에는 2차원 배열로 입력을 받아 회의가 끝나는 시간 순서대로 정렬을 하여 문제를 풀이하였다.

채점을 하는데 80퍼 쯤에서 틀려서 고민을 해보니 다음과 같은 반례가 있었다.5
4 4
4 4
3 4
2 4
1 4이렇게 되면 결과가 3이 나와야하는데 2가 나오는 것이였다.끝나는 시간이 같으면 시작하는 시간 순서대로 정렬을 해줘야 정답이 제대로 출력이 된다.

 

소스코드.

import sys

N = int(sys.stdin.readline())

meeting = []
for i in range(N):
    s, e = map(int, sys.stdin.readline().split())
    meeting.append([s, e])

meeting = sorted(meeting, key=lambda item: (item[1], item[0])) # 끝나는 시간, 시작하는 시간 정렬
cnt = 0
now = 0
for start, end in meeting:
    if start >= now: #현재 시간보다 시작시간이 같거나 크다면
        now = end # 시작시간을 끝나는 시간으로 변경
        cnt+= 1
print(cnt)