[Python - 백준 1931] 회의실 배정

문제 설명

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

입력형식

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

출력형식

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

입출력 예시

풀이 과정

이 문제의 핵심은 회의를 가장 많이 하려면 가장 일찍 끝나는 회의를 회의실에 넣어야한다는 것이다.

그래서 나는 가장 일찍 끝나는 회의순으로 정렬을 해주었다. 만약 끝나는 시간이 같은 것이 존재한다면? 시작시간이 빠른 순으로 정렬하게 하였다. 그 다음에 순차적으로 list를 검사하며, start가 end보다 크거나 같은 경우에만 end를 넣어주는 식으로 for문을 검사하면 되는 문제였다.

def solution(arr):
    answer = 0
    arr.sort(key=lambda k: (k[1], k[0]))
    end = 0
    for start, done in arr:
        if start >= end:
            answer += 1
            end = done
    return answer


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

print(solution(arr))