[BOJ/Python] 1931번: 회의실 배정

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

문제

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

입력

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

출력

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

---

구상

이 문제의 키 포인트는 정렬을 어떻게 할지이다. 여기서 포인트는 최대한 많은 회의실을 잡으려면, 빨리 끝나는 애들을 우선으로 잡아야된다 이다. 그래서 최종 정렬은 끝나는 시간대로 오름차순으로 정렬되어있어야되고, 그 안에서 시작하는 시간대로도 오름차순 되어있어야 더 효율적인 계산이 가능하다

코드

N=int(input())
arr=[]
count=0
for _ in range(N):
    X=list(map(int,input().split()))
    arr.append(X)

arr.sort(key=lambda x: x[0])
arr.sort(key=lambda x: x[1])
time=arr[0][0]
for i in range(N):
    if time <= arr[i][0]:
        count+=1
        time = arr[i][1]

print(count)

우선 시작하는 시간으로 정렬 후 끝나는 시간으로 정렬해야 최종적으론 빨리 끝나는 애들이 우선적으로 정렬되어있을 것이다! 배열 인덱스로 정렬하는 법은 구글링을 통해 key=lambda라는 것을 사용한다는 것을 알게되었다.