프로그래머스 '입국 심사' 문제풀이 - Henry's Algorithm

 

 

 

문제 설명

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

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

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

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

제한사항

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

입출력 예

ntimes     return

6

[7, 10]

28

입출력 예 설명

가장 첫 두 사람은 바로 심사를 받으러 갑니다.

7분이 되었을 때, 첫 번째 심사대가 비고 3번째 사람이 심사를 받습니다.

10분이 되었을 때, 두 번째 심사대가 비고 4번째 사람이 심사를 받습니다.

14분이 되었을 때, 첫 번째 심사대가 비고 5번째 사람이 심사를 받습니다.

20분이 되었을 때, 두 번째 심사대가 비지만 6번째 사람이 그곳에서 심사를 받지 않고 1분을 더 기다린 후에 첫 번째 심사대에서 심사를 받으면 28분에 모든 사람의 심사가 끝납니다.

출처

※ 공지 - 2019년 9월 4일 문제에 새로운 테스트 케이스를 추가하였습니다. 도움을 주신 weaver9651 님께 감사드립니다.

 

 

문제 푸는 방법:

이진 탐색을 사용하여 문제를 푼다.

1분 부터 (가장 오래걸리는 심사관의 입국 심사 시간)*n 까지 범위를 두고,

mid = (left + right) / 2

mid라는 시간에서 각 심사관이 몇 명을 심사볼 수 있는지 isPassed 라는 메소드를 통해 확인

isPassed -> true 이면 현재 시간(mid)에 사람들이 모두 심사를 볼 수 있음(하지만 우리는 최솟값을 얻기 위한 것이니, left <= right일때까지 반복 true이면 right = mid -1로 더 줄여나감

isPassed -> false이면 현재 시간(mid)에 사람들이 모두 심사를 볼 수 있는 건 아님 => left = mid+1 로 현재시간을 더 늘려보기로 함

 

해당 과정을 left <= right 일때까지 반복하고 return 한다.

 

정답 코드(Java):

import java.util.*;

class Solution {
    public long solution(int n, int[] times) {
        Arrays.sort(times);
        return BinarySearch(times,n,times[times.length - 1]);
    }
    
    long BinarySearch(int[] times, int n, long max){
        long left = 1, right = max * n;
        long mid = 0;
        long ans = Long.MAX_VALUE;
        
        while (left <= right){
            mid = (left+right)/2;
            
            if (isPassed(times,n,mid)){
                ans = ans > mid ? mid : ans;
                right = mid - 1;
            }else{
                left = mid + 1;
            }
        }
        return ans;
    }
    
    boolean isPassed(int[] times, int n, long mid){
        long amount = 0;
        
        for(int i=0; i<times.length; ++i){
            amount += mid/times[i];
        }
        if (amount>=n) return true;
        else return false;
    }
}