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

(백준) 1931 - 회의실 배정

문제

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

입력

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

출력

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

코드

import sys

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

arr = []
for _ in range(N):
    arr.append(list(map(int,sys.stdin.readline().split())))

arr.sort(key = lambda x: (x[1], x[0]))
end_time = 0
cnt = 0
for i in arr:
    if end_time <= i[0]:
        cnt += 1
        end_time = i[1]

print(cnt)

풀이

N개의 회의의 시작시간과 종료시간이 차례대로 공백을 두고 여러개 들어온다.

따라서 리스트를 하나 정의하고 입력받은 값들을 int값들이 담긴 리스트에 추가(append)한다.

arr = []

for _ in range(N):
    arr.append(list(map(int,sys.stdin.readline().split())))

그 다음 정렬을 해주는데 그 이유는

회의가 가장 많이 들어가기 위해서 회의가 끝나는 시간을 기준으로 정렬을 해 준다.

그리고 끝나는 시간이 같은 회의에 대해서는 시작 시간을 기준으로 정렬을 해 준다.

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

이는 sort 함수의 key 함수를 한번 읽어 보면 좋을 것 같다. key를 이용한 정렬은 자주 나온다.