백준 코딩테스트/그리디(Greedy)

1931번: 회의실 배정(8 / 2) - 실버 1

monkeykim 2024. 10. 7. 00:28

문제

한 개의 회의실이 있는데 이를 사용하고자 하는 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

코드

import sys

n = int(input())

times = [tuple(map(int, sys.stdin.readline().split())) for _ in range(n)]

# 종료 시간을 기준으로 정렬
# lambda x: x를 매개변수로 가지는 람다 함수를 선언.
# x[1]: x[1]를 기준으로 오름차순 정렬을 한다.
# x[0]: x[1]이 동일할 경우 x[0]을 기준으로 오름차순 정렬을 한다.
times.sort(key=lambda x: (x[1], x[0]))

count = 0
end_time = 0

for time in times:
    if time[0] >= end_time:
        end_time = time[1]
        count += 1

print(count)

코드 풀이

  1. n개의 개수만큼 tuple 타입의 input을 n개의 개수만큼 입력받아 리스트에 추가한다.
    • input대신 sys.stdin.readline() 으로 입력을 받으면 시간에 유리하다.
  2. 회의의 최대의 개수를 구하는 것이므로 종료시간을 기준으로 오름차순 정렬을 해준다.
    • 문제 중에 "한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다." 를 고려하여 종료시간을 기준으로 정렬할 때 동일한 종료 시간이 있을 때는 시작시간을 기준으로 오름차순 정렬을 한다.
    • list.sort(key=lambda) 를 사용하여 람다 함수를 사용하여 정렬을 할 수 있다.
  3. 회의시간을 반복하면서, 시작시간이 종료시간보다 크거나 같은 경우, 종료시간을 시작시간의 종료시간으로 변경하고 count를 늘리면서 반복을 진행한다.

핵심 포인트

  • 그리디 알고리즘과 정렬은 아주 밀접한 관계에 있음
    • 종료시간을 기준으로 오름차순 정렬을 함
    • 종료시간이 동일한 경우 시작시간을 기준으로 오름차순 정렬을 함
    • list.sort(key=lambda) 를 사용
  • 시작시간이 종료시간보다 크거나 같은 경우 종료시간을 교체해주는 로직