<Level 3> 입국심사 with 파이썬

 

문제 설명

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

 

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

 

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

 

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

제한사항

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

def solution(n, times):
    answer = 0
    start = 0
    end = max(times) * n
    while start <= end :
        mid = (start + end)//2
        total = 0
        for time in times:
            total += mid//time
        if total >= n :
            answer = mid
            end = mid - 1
        else :
            start = mid + 1
    return answer

 

 - 예시를 보면 답인 28에서 times의 원소들을 각각 나눈 몫들의 합이 n과 일치함을 알 수 있다.

 

 - 예시로 다시 설명을하자면 먼저 최대로 걸릴 수 있는 시간은 times의 최댓값과 n을 곱한 값이다. 10분걸리는 심사관이 6명을 혼자 심사하면 60분이 걸린다. 만약 40분정도가 걸린다면 어떤식으로 심사가 이루어질까? 7분 걸리는 사람은 40//7 인 5명을 처리할 수 있고, 10분걸리는 사람은 40//10 인 4명을 처리할 수 있다. 즉 40분이면 총9명이나 처리할 수 있으니 이 보다 더 적은 시간으로도 6명을 처리할 수 있다.

 

 - 이제 위와 같은 방식으로 이분탐색하여 6명이상을 처리할 수 있는 시간을 탐색하면된다.

 

 - 만약 times의 원소들을 나눈 몫의 합이 n보다 크다면 더 적은시간으로도 가능하다는 의미이므로 end를 mid-1로 줄인다. 그러나 n보다 작다면 더 큰시간이 필요하다는 의미이므로 start를 mid+1로 늘려나간다.

 

 

---

 

 

 

참고

 

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