[프로그래머스] '입국심사' 알고리즘 풀이 - Python

Contents

1. 문제 설명
   
   [제한사항]
   
   [입출력 예]
2. 알고리즘 분석 
   
   [나의 풀이]
   
   [Most 1 의 풀이]

---

 

문제 설명

 

n명이 입국심사를 위해 줄을 서서 기다리고 있습니다. 각 입국심사대에 있는 심사관마다 심사하는데 걸리는 시간은 다릅니다.

 

처음에 모든 심사대는 비어있습니다. 한 심사대에서는 동시에 한 명만 심사를 할 수 있습니다. 가장 앞에 서 있는 사람은 비어 있는 심사대로 가서 심사를 받을 수 있습니다. 하지만 더 빨리 끝나는 심사대가 있으면 기다렸다가 그곳으로 가서 심사를 받을 수도 있습니다.

 

모든 사람이 심사를 받는데 걸리는 시간을 최소로 하고 싶습니다.

입국심사를 기다리는 사람 수 n, 각 심사관이 한 명을 심사하는데 걸리는 시간이 담긴 배열 times가 매개변수로 주어질 때, 모든 사람이 심사를 받는데 걸리는 시간의 최솟값을 return 하도록 solution 함수를 작성해주세요.

 

 

 

제한사항

• 입국심사를 기다리는 사람은 1명 이상 1,000,000,000명 이하입니다.
• 각 심사관이 한 명을 심사하는데 걸리는 시간은 1분 이상 1,000,000,000분 이하입니다.
• 심사관은 1명 이상 100,000명 이하입니다.

 

입출력 예

n	times	return
6	[7, 10]	28

알고리즘 분석

 

• 나의 풀이

# 시간 초과 
import heapq as hq

def solution(n, times):
    times = [(time, time) for time in times]
    hq.heapify(times)
    for i in range(n):
        time = hq.heappop(times)
        time = (time[0]+time[1], time[1])
        hq.heappush(times, time)
    answer = list(map(lambda x: x[0] - x[1], times))
    return max(answer)

 

for 문을 통해서 n만큼 반복하는 방법입니다. 

간단한 방법이지만 문제점이 있습니다. 

n의 최댓값이 무려 1,000,000,000이라는 점입니다. 

때문에 시간초과가 발생할 것이라고 예상했는데 역시나 시간 초과가 발생했습니다. 

 

 

• 이분탐색 풀이를 참고한 코드

# C++을 이용한 깔끔한 풀이가 있어서 파이썬으로 따라해봤습니다.
# 이분탐색 
def solution(n, times):
    times.sort()  # times를 오름차순으로 정렬 
    left = (times[0] * n) // len(times) # 나올 수 있는 가장 작은 값 
    right = times[-1] * n  # 나올 수 있는 가장 큰 값
    while left < right:
        mid = (left + right) // 2 # 가장 큰, 작은 값의 중간 값 
        temp = 0  # temp: 중간 값으로 처리할 수 있는 n의 수 
        for time in times:
            temp += mid // time # mid의 시간동안 time당 mid//time만큼 처리 가능 
        if temp >= n: # temp가 n과 같거나 클 경우 
            right = mid # right를 mid가 대체 
        else: # 아닐 경우 
            left = mid + 1 # left를 mid + 1이 대체 
    return right

 

 

코딩테스트 연습 - 입국심사