[프로그래머스] 입국심사,python

문제설명

programmers.co.kr/learn/courses/30/lessons/43238

n명이 입국심사를 위해 줄을 서서 기다리고 있습니다. 각 입국심사대에 있는 심사관마다 심사하는데 걸리는 시간은 다릅니다.
처음에 모든 심사대는 비어있습니다. 한 심사대에서는 동시에 한 명만 심사를 할 수 있습니다. 가장 앞에 서 있는 사람은 비어 있는 심사대로 가서 심사를 받을 수 있습니다. 하지만 더 빨리 끝나는 심사대가 있으면 기다렸다가 그곳으로 가서 심사를 받을 수도 있습니다.
모든 사람이 심사를 받는데 걸리는 시간을 최소로 하고 싶습니다.
입국심사를 기다리는 사람 수 n, 각 심사관이 한 명을 심사하는데 걸리는 시간이 담긴 배열 times가 매개변수로 주어질 때, 모든 사람이 심사를 받는데 걸리는 시간의 최솟값을 return 하도록 solution 함수를 작성해주세요.

풀이

처음에는 for 문으로 직접 시간을 찾는 방식으로 풀었는데, time out으로 문제를 제대로 해결하지 못했다.

그래서 다른 사람들의 풀이를 찾아서 이분탐색으로 풀어보았다. 

검사에 걸리는 총 시간을 left right로 두고, 그 시간 안에서 최적의 시간을 찾아나가는 방식으로 문제를 풀어보았다.

 
def solution(n, times):
    ## 0 ~ 최악의 시간까지 탐색
    left , right = 0, max(times) * n
    answer =0
    while left <= right:
        mid = (left + right) //2
        ## 각 심사관은 이 시간동안 총 몇명을 받을 수 있을까?
        people = 0
        for inspector in times:
            people += (mid//inspector)
            
            if people >= n:
                break
        ## 시간내에 다 받을 수 있었으면, 값을 할당하고 범위를 줄인다
        if people>=n:
            right = mid-1
            answer = mid
        ## 시간내에 다 받지 못했으면, 시간을 늘린다
        else:
            left = mid+1
            
    return  answer