백준 1931번 회의실 배정 (Python, Greedy, Silver 1)

문제 이해

문제

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

입력

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

출력

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

문제 접근

회의를 최대한으로 많이 하려면 어떻게 해야할까?

1. 회의 시작시간을 기준으로 가장 빠른 회의를 선택
   이 경우에는 만약 회의시간이 많이 길다면 다른 회의를 선택하지 못하는 경우가 발생하므로 옳지 못하다!
2. 회의의 기간을 짧은 기준으로 선택
   회의의 기간을 짧은 기준으로 선택한다면 시작 시간을 알지 못하고 끝시간을 알지 못함. 이렇게 sort된 배열을 확인할 때 시작시간과 종료시간이 뒤죽박죽이 된다.
3. 회의의 종료시간을 기준으로 가장 빠른 회의를 선택
   종료시간을 기준으로 선택한다면 종료시간을 기준으로 sort하고 이렇게 sort된 배열을 순차적으로 확인하면서 1번 회의가 종료되었다면 다음 회의시간중 가장 빨리 끝나는 회의를 자동 선택하는 것이므로 옳다!

따라서 3번 방법을 선택한다.

문제 풀이

끝나는 시간을 기준으로 정렬하기

lambda를 사용하여 정렬해준다.

for i in range(N):
    start, end = map(int,sys.stdin.readline().split())
    timeline.append((start, end))
timeline.sort(key=lambda x : (x[1],x[0]))

순차적으로 선택하기

첫번째 회의는 필수적으로 선택한다 그리고 2번째 회의부터 끝나는 시간이 시작시간과 비교했을 때 시작시간이 클 경우에 그 회의를 선택하고 회의의 개수를 1 증가시킨다.

count = 1
end = timeline[0][1]
for i in range(1, N):
    if timeline[i][0]>=end:
        end = timeline[i][1]
        count += 1

제출 코드

import sys
N = int(sys.stdin.readline())
timeline = []
for i in range(N):
    start, end = map(int,sys.stdin.readline().split())
    timeline.append((start, end))

timeline.sort(key=lambda x : (x[1],x[0]))
count = 1
end = timeline[0][1]
for i in range(1, N):
    if timeline[i][0]>=end:
        end = timeline[i][1]
        count += 1

print(count)