[백준] 1931 회의실 배정(실버2) - Python

jiiiii·2022년 3월 30일
0

알고리즘

목록 보기
15/19

문제

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

입력

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

출력

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


풀이

처음에는 그냥 시작 시간에 맞춰 정렬해서 넣으면 되지 않을까? 했는데 그렇게 되면 시작하는 시간이 가장 빠른 회의가 하루 종일 회의를 하기로 예약이 되어있다면 그 날은 회의실에서 회의를 한 개밖에 할 수 없게 된다.

그래서 끝나는 시간을 기준으로 먼저 정렬을 하고, 그 다음으로 시작 시간을 기준으로 정렬을 하게 되면 한 회의가 시간을 다 잡아먹지 않고 최대한 많은 회의를 할 수 있다고 생각했다.

meetings = []
for _ in range(n):
    s, e = map(int, input().split())
    meetings.append((s, e))

먼저 meetings에 회의 시작/끝 시간을 모두 담고

meetings.sort(key=lambda x: (x[1], x[0]))

lambda를 사용하여 sort의 정렬 기준을 종료 시간시작 시간으로 잡는다

cnt, last_end = 0, 0
for start, end in meetings:
    if start >= last_end:
        cnt += 1
        last_end = end

print(cnt)

그리고 meetings를 돌면서 확인하고 있는 회의의 시작 시간이 현재 진행되고 있는 회의의 종료 시간 이후라면 cnt를 하나 늘리고 해당 회의의 종료 시간을 last_end에 넣는 것을 반복한다.

그리고 for loop를 다 돌면 cnt를 프린트하면 끝!

전체 코드

meetings = []
for _ in range(n):
    s, e = map(int, input().split())
    meetings.append((s, e))

meetings.sort(key=lambda x: (x[1], x[0]))

cnt, last_end = 0, 0
for start, end in meetings:
    if start >= last_end:
        cnt += 1
        last_end = end

print(cnt)

0개의 댓글