[프로그래머스 / 이분탐색] 입국심사 (java) ⭐️

[문제]

문제 설명

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

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

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

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

제한사항

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

입출력 예

ntimesreturn

6

[7, 10]

28

입출력 예 설명

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

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

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

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

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

출처

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

 

 

[문제 접근]

처음에 어떻게 풀어야 할지 감이 잘 안왔다. (알고리즘 제외하고 그냥 나한테 풀으라고 해도 어떻게 접근해야할지 감이 안왔던..)

그래서 구글 서치를 통해 힌트를 얻었고, 문제를 이해하게 되었다. 그리고 어떻게 이분 탐색을 이용해야 되는지에 대한 힌트도 얻었다. 

 

핵심은 

추정 시간값 / 각 심사관 별 심사시간 = 심사관 당 맡을 수 있는 입국자 수 

라는 것이다. 

 

이 식을 이해하고 나니 문제에 대한 이해를 할 수 있게 되었다. 

 

[코드] - 1차 실패

import java.util.*;

class Solution {
    public long solution(int n, int[] times) {
        long answer = 0;
        int d = times.length;
        
        Arrays.sort(times); // 크기 순으로 정렬
        
        int left = 1;
        int right = d * times[d-1]; // 가장 오래 걸리는 심사관이 n명 보는거- 최대
        int sum = 0;
        
        while(left <= right){
            int mid = (left + right) / 2;
            sum = 0;
            for(int i = 0; i < times.length; i++){
                sum += mid / times[i];
            }
            if(sum < n){
                left = mid + 1;
                // 더 맡을 수 있으니까?
            }else {
                right = mid -1;
                answer = right;
            }
        }
        
        return answer;
    }
}

answer가 계속 0이 나와서 고민했는데, answer의 타입이 문제였다!!!!! 코드의 효율성 통과를 위해 (...) 다른 변수들도 타입을 long으로 바꿔준다.

하지만 그래도 틀렸다. 최댓값을 잡기 위해선 answer에서 right이 아니라 mid 값을 잡아주어야 한다. 

 

[코드2] 

근데 그래도 테스트 케이스 절반 통과 못함 ..

import java.util.*;

class Solution {
    public long solution(int n, int[] times) {
        long answer = 0;
        
        Arrays.sort(times); // 크기 순으로 정렬
        
        long left = 1;
        long right = n * times[times.length-1]; // 가장 오래 걸리는 심사관이 n명 보는거- 최대
        long sum = 0;
        
        while(left <= right){
            long mid = (left + right) / 2;
            sum = 0;
            for(int i = 0; i < times.length; i++){
                sum += mid / times[i];
            }
            if(sum < n){
                left = mid + 1;
                // 더 맡을 수 있으니까?
            } else{
                right = mid -1;
                answer = mid; // 수정
            }
        }
        
        return answer;
    }
}

문제점 : 

n = 5이고, [4, 3, 1, 1, 1]이면 첫번째 5명은 바로 심사를 받으므로 정답이 4가 되어야 하나?.. 아니면 2가 되어야 하나

 

[코드3] - 성공코드

(아..빡치네;..) right의 초기값에서 times에 대하여 long 타입으로 변환이 필요했다. (int형 값의 범위를 넘어가기 때문에)

자료형 수정하니 테스트케이스 전부 통과!

import java.util.*;

class Solution {
    public long solution(int n, int[] times) {
        long answer = 0;
        
        Arrays.sort(times); // 크기 순으로 정렬
        
        long left = 1;
        long right = n * (long)times[times.length-1]; // 가장 오래 걸리는 심사관이 n명 보는거- 최대
        //타입
        long sum = 0;
        answer = right;
        while(left <= right){
            long mid = (left + right) / 2;
            sum = 0;
            for(int i = 0; i < times.length; i++){
                sum += mid / times[i];
            }
            if(sum < n){
                left = mid + 1;
                // 불가능-> 시간을 더 많이
            }
            else{  
                right = mid -1;
                answer = mid;
                //가능 -> 시간을 최대한 줄여보기
                
            }
        }
        
        return answer;
    }
}

 

 

 

[참고]

https://woongsin94.tistory.com/185

프로그래머스_입국심사

[문제 링크]

https://programmers.co.kr/learn/courses/30/parts/12486

프로그래머스