본문 바로가기
코딩 테스트

[백준 1931번] 회의실 배정 파이썬

by 얀나대장 2023. 4. 14.

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이다.

출력

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


해결법

최대한 많은 회의를 해야하므로 회의가 끝나는 시간 순대로 정렬한 후, 앞으로 있을 회의 시작시간이 현재 끝나는 시간보다 크면 회의를 update한다.

 

import sys; input = sys.stdin.readline

n= int(input())
#output = 회의실을 사용할 수 있는 최대 개수

#시작시간=끝나는 시간 ok

#시간 수가 너무 크기때문에 dp로 만들 수 없음!
table = []

for _ in range(n):
    start, end = map(int, input().split())
    # 일찍 끝나는 시간대로 정렬
    table.append([end,start])
table=sorted(table)

#제일 작은 값으로 회의!
meeting = table[0]
output = 1
for i in range(n-1):
    #끝나는 시간이 누군가의 시작시간이면 +1
    if meeting[0]<=table[i+1][1]:
        output +=1
        #회의실 선택
        meeting = table[i+1]
print(output)

 

# case2

lamba를 이용해 뒤의 변수로 정렬

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

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

time = sorted(time, key=lambda a: a[1]) # 끝나는 시간을 기준으로 오름차순 정렬

meeting = 0 # 회의의 마지막 시간을 저장할 변수
output = 0 # 회의 개수를 저장할 변수

for i, j in time:
  if i >= meeting: # 시작시간이 회의의 마지막 시간보다 크거나 같을경우
    conut += 1
    meeting = j

print(output)

배운점

회의를 가장 많이 하려면 끝나는 시간이 짧을 수록 좋다는 것을 배웠다.

 

복습

리스트에서 key를 사용해 오름차순 순서 정하기

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

 

댓글