본문 바로가기
백준/그리디

[백준] #1931 회의실 배정 (python)

by 똥먹는낙타 2022. 3. 5.
728x90
반응형

https://www.acmicpc.net/problem/1931

 

1931번: 회의실 배정

(1,4), (5,7), (8,11), (12,14) 를 이용할 수 있다.

www.acmicpc.net

 

문제

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

# Code

n = int(input())
time = []
for i in range(n):
    start, end = map(int, input().split())
    time.append([start, end])

time=sorted(time, key=lambda x: (x[1],x[0])) # 회의가 끝나는 시간으로 오름차순 정렬 후 시작 시간이 빠른 것으로 오름차순 정렬

count = 1
end_time = time[0][1]
for i in range(1,n):
    if time[i][0] >= end_time:
        count+=1
        end_time = time[i][1]

print(count)

# Comment

처음에 회의 시작시간으로 정렬한 후 끝나는 시간이 빠른 것부터 정렬해서 풀어보려고 했는데 이렇게 하면 좋은 풀이가 딱히 떠오르지 않아서 고민했다. 결국 어떻게 정렬해야하는지 약간 참고했다.

빠르게 끝나는 회의일수록 더 많은 회의가 진행될 수 있으므로, 회의가 끝나는 시간이 빠른 것부터 우선적으로 오름차순 정렬을 해준다. 그리고 회의가 끝나는 시간과 다음 회의가 시작하는 시간이 가까울수록 더 많은 회의를 진행할 수 있기 때문에 회의 시작 시간이 빠른 것부터 오름차순 정렬을 해준다.

(1 4)  (1 2)  (2 2)  (2 5)  4개의 회의가 있다고 할 때,

처음 기준인 회의가 끝나는 시간이 빠른 것부터 정렬하면 (2 2) (1 4) (3 5) (2 5) 가 되고, 두번째 기준인 끝나는 시간이 같을 경우 회의 시작 시간이 빠른 것부터 정렬하면 (1 2) (2 2) (1 4) (3 5) 가 되어서 처음에 (1 2) 회의를 진행할 수 있고, 그 다음으로는 (2 2), (3 5) 총 3개의 회의를 진행할 수 있다.

* 여기서 정렬할 때 lambda 함수를 사용했다. time=sorted(time, key=lambda x: (x[1],x[0])) 는 time[1] 값으로 먼저 정렬한 후 time[0] 값으로 정렬한다는 뜻이다. 알아두면 유용하게 많이 쓸 듯 하다.

https://offbyone.tistory.com/73

 

파이썬 문법 5 - 람다(lambda) 함수

파이썬에서 "lambda" 는 런타임에 생성해서 사용할 수 있는 익명 함수 입니다. 이것은 함수형 프로그래밍 언어에서 lambda와 정확히 똑같은 것은 아니지만,  파이썬에 잘 통합되어 있으며 filter(), map

offbyone.tistory.com

여기서 람다에 대해서 참고했다!

728x90
반응형

댓글