[Python] 1931번 : 회의실 배정

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

1931번: 회의실 배정

문제

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

입력

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

출력

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

 

풀이

회의가 끝나는 시점에 대해 시간을 저장한 배열을 정렬한다.

순차적으로 회의를 확인하면서 배정된 회의실의 회의가 끝나는 시점보다 회의 시작 시간이 크거나 같은 회의를 다음 회의실에 저장합니다. (배정된 회의실의 회의가 끝나는 시점보다 먼저 시작하는 회의는 배정될 수 없음!)

 

보완한 점 : 회의가 끝나는 시점에 대해 배열을 정렬했는데 끝나는 시점이 같은 경우에 회의 시작을 먼저하는 쪽이 앞쪽에 위치하도록 sorted의 lambda key에 x[1]에서 x[0]까지 추가해야한다.

n = int(input())
arr = [list(map(int,input().split())) for _ in range(n)]

arr = sorted(arr, key = lambda x : (x[1],x[0]))

count = 1
end = arr[0][1]

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

print(count)