백준 1931번: 회의실 배정 (G5) - Python 풀이

문제 설명

https://www.acmicpc.net/problem/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

풀이 과정

회의실을 사용할 수 있는 회의의 최대 개수를 구해야 하는 문제입니다. 어떤 기준으로 정렬해야 하는지만 잘 생각하면 풀립니다.

아 뭔가 특정 알고리즘을 생각나지는 않고 기준만 잘 정해서 정렬만 잘 하면 될 것 같은데… ← 그리디의 기운. 여러 회의 중에 어떤 회의를 고르는 게 가장 좋을까요? 바로 빨리 시작하면서 빨리 끝나는 회의여야 합니다.

그렇담 회의 시작 시간 + 회의 진행 시간 의 값이 가장 작은 것부터 선택하면 되겠다고 생각했는데 결국 이 값은 회의 끝나는 시간의 값과 같습니다. ex) 입력값 1 4 ⇒ 시작 시간 1 + 회의 진행 시간 3 = 4 (회의 끝나는 시간)

그래서 회의 종료 시간을 기준으로 정렬하고, 만약 종료 시간이 같다면 시작 시간이 같은 것부터 하는 게 더 많은 회의를 진행할 수 있으니까 시작 시간을 기준으로 정렬했습니다.

이는 파이썬 sort에 람다함수로 설정하면 됩니다. meetings.sort(key=lambda x: (x[1], x[0]))

하나의 회의를 진행한다면 이후에 선택될 수 있는 회의는 해당 회의의 종료 시간 이후에 시작하는 회의여야 합니다. 따라서 if 문을 통해 필터링을 거쳐 다음 회의를 선택했습니다.

반복문을 통해 회의를 선택하고, 선택할 때마다 answer 값을 카운트 증가시켜서 답을 구했습니다.

정답 코드

N = int(input())
meetings = []

# 회의 정보 입력
for _ in range(N):
    meetings.append(tuple(map(int, input().split())))

# 회의 종료 시간을 기준으로 정렬, 종료 시간이 같을 때는 시작 시간을 기준으로 정렬
meetings.sort(key=lambda x: (x[1], x[0]))

answer = 0
start_time = 0
# 반복문을 통해 모든 회의를 확인
for i in range(N):
    # 현재 회의의 종료 시간 이후에 시작하는 회의만 필터링
    if meetings[i][0] >= start_time:
        start, end = meetings[i]
        # 필터링 기준이 되는 값을 현재 회의의 종료 시간으로 업데이트
        start_time = end
        # 카운트 증가
        answer += 1

print(answer)

이상입니다.