백준(Baekjoon) 1931번 회의실 배정 문제 풀이 | 파이썬 Python, 그리디 알고리즘

회의실 배정

출처: www.acmicpc.net/problem/1931

1931번: 회의실 배정

문제

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

입력

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

출력

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

예제 입력1

11
1 4
3 5
0 6
5 7
3 8
5 9
6 10
8 11
8 12
2 13
12 14

예제 출력1

4

풀이

n=int(input())
lst=[]
for i in range(n):
    lst.append(list(map(int,input().split())))
lst=sorted(lst,key=lambda x:x[0])
lst=sorted(lst,key=lambda x:x[1])
last,result=0,0
for j in lst:
    if j[0]>=last:
        result+=1
        last=j[1]
print(result)

문제를 풀기 위해서 먼저 고려해야하는 것은 시작하는 시간이다. 따라서 sorted를 사용하여 시작시간을 기준으로 정렬하였다. 그리고 또 고려해야하는 것은 끝나는 시간이다. 최대한 회의실의 사용시간이 짧을수록 회의의 개수가 늘어날 것이다. 따라서 시작시간을 기준으로 정렬한 리스트를 끝나는 시간을 기준으로 한번 더 정렬해준다. 여기까지의 과정을 찾아냈다면, 나머지 정답을 구하는 과정은 쉽게 구현할 수 있다.

끝나는 시간을 기억해줄 변수 last와 결과값을 저장할 result를 선언해준다. 그리고 정렬한 리스트를 순차적으로 탐색하며 시작하는 시간이 last의 값보다 큰 경우 result를 1 더해주고, last 변수에 해당 끝나는 시간을 넣어준다.