코테 공부/백준

[Python/그리디] 1931 회의실 배정

prefer_all 2023. 3. 7. 16:19

문제

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

입력

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

출력

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

예제 입력 예제 출력
11
1 4
3 5
0 6
5 7
3 8
5 9
6 10
8 11
8 12
2 13
12 14
4

풀이

각 단계에서 가장 최적의 답을 찾는 그리디 알고리즘의 문제이다.
처음에는 '회의 시간이 얼마나 긴가'를 기준으로 접근하려고 했는데 
이 문제는 회의를 가장 많게 배치하는 것이기 때문에 정렬로 접근해야 한다.

일찍 끝나는 회의를 먼저 배치하고 끝나는 시간이 같은 회의들에 대해서는 회의시간이 짧은(시작 시간이 늦은) 회의를 우선순위로 한다.

'''
한 개의 회의실을 N개의 회의에 대해 사용
겹치기 않게 하면서 사용할 수 있는 회의의 최대 개수
하나가 끝남과 동시에 시작할 수 있음
진짜 아이디어조차 떠오르지 않는다
앞에서부터 선택하면서 모든 경우의 수 고려하는 것 아님
정렬: 회의 시간 기준 아님
1. 끝나는 시간 오름차순
2. 회의 끝나는 시작이 같을 경우 더 늦게 시작하는 순서(시작 시간 오름차순)
'''
#import sys
#sys.stdin = open('./input.txt', 'r')

N = int(input())
arr = []
for _ in range(N):
    start, end = map(int, input().split())
    arr.append([start, end])
#print(arr)
arr = sorted(arr, key = lambda x: (x[1], x[0]))
start, end = arr[0][0], arr[0][1]
cnt = 1
for i in range(1, len(arr)):
    if arr[i][0] >= end:
        cnt += 1
        start, end = arr[i][0], arr[i][1]
print(cnt)

 

Lambda 이용해서 정렬하기

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