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

1931번: 회의실 배정

 

문제

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

 

입력

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

 

출력

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

 

n = int(input())
data = [list(map(int, input().split())) for _ in range(n)]
data.sort(key = lambda x: (x[1], x[0]))
cnt = 1
finish = data[0][1]
for i in range(1, n):
    if data[i][0] >= finish:
        cnt += 1
        finish = data[i][1]
print(cnt)

이 문제는 회의에 걸리는 시간을 기준으로 나열하고, 회의 시작 시간을 기준으로 나열하는 것이 중요합니다. 회의에 걸리는 시간이 가장 짧을 때 가장 많은 회의를 할 수 있기 때문입니다. 또한 회의 시작 시간이 같은 경우도 있기 때문에 회의 시작 시간의 순서도 고려해주어야 합니다.

- sort()에 람다를 이용해서 두 번째 원소(회의에 소요되는 시간)를 기준으로 나열하고 첫 번째 원소(회의 시작 시간)을 기준으로 나열해줍니다.

- 그 다음 두 번째 리스트 요소부터 탐색하기 위해 cnt를 1로 초기화 하고, 회의 종료 시간(finish)을 첫 번째 원소의 회의 소요 시간으로 초기화 해줍니다.

- data 리스트의 원소 중 회의 시작 시간이 finish보다 크거나 같다면 cnt를 하나 증가시켜주고, finish를 해당 data 리스트의 회의 소요 시간으로 바꿔줍니다.

- 위의 과정을 n - 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
3 5
0 6
5 7
3 8
5 9
6 10
8 11
8 12
2 13
12 14

# cnt = 1, finish = 4로 시작
# i = 1~2, if문이 실행되지 않음
# i = 3, cnt = 2, finish = 7
# i = 4~6, if문이 실행되지 않음
# i = 7, cnt = 3, finish = 11
# i = 8~9, if문이 실행되지 않음
# i = 10, cnt = 4, finish = 14
# 4 출력

피드백은 언제나 환영합니다.