[프로그래머스 Level.3] 입국심사 (연습문제) (Java)

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/43238

프로그래머스

 

 

 

 

코딩테스트 연습 > 이분탐색 > 입국심사

 

 

 

 

문제 설명

 

 

 

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

 

 

입출력 예 설명

 

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

 

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

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

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

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

 

 

 

 

 

 

나의 코드

import java.util.*;

class Solution {
    public long solution(int n, int[] times) {
        long answer = 0;
        
        Arrays.sort(times);
        
        // target => 모든 사람이 심사를 받는데 걸리는 시간
        long start = 1;  // 최선의 경우 걸리는 시간
        long end = (long) times[times.length-1] * n;  // 최악의 경우 걸리는 시간
        
        while(start <= end) {
            long mid = (start + end) / 2;
            long sum = 0;  // mid의 시간 동안 총 심할 수 있는 인원의 합
            
            // mid / time = 각 심사위원이 검사 할 수 있는 인원수
            for(int time : times) sum += mid / time;
            
            if(sum >= n) { // 심사해야 할 인원보다 많이 함 -> 시간을 줄여서 더 최선의 경우인 시간 만듬
                answer = mid;
                end = mid - 1;
            } else { // 심사해야 할 인원보다 덜 함 -> 시간 더 필요
                start = mid + 1;
            }
        }
        
        return answer;
    }
}

 

풀이

1. 모든 사람이 심사를 받는데 걸리는 시간이 상당히 클 수 있으므로(1,000,000,000명을 1,000,000,000시간동안 심사 할 수 있기 때문에) 이분 탐색(Binary Search)을 이용하여 탐색 범위를 절반으로 줄여가며 푸는 문제이다.
2. 우선 심사위원들이 심사를 하는데 걸리는 시간의 배열인 times를 오름차순으로 정렬한다.
3. 모든 사람이 심사를 받는데 걸리는 시간을 구하는 문제이므로 target을 모든 사람이 심사를 받는데 걸리는 시간으로 하고 start를 최소의 경인 1로 end를 최대의 경우인 제일 심사하는데 오래 걸리는 시간인 times[times.length-1]에 n을 곱한 값으로 한다. (수가 상당히 클 수 있으므로 long형으로 생성한다)
4. 이제 start가 end 이상이 되면 종료하는 while문을 생성하여 가능 불가능에 따라 범위를 절반으로 줄여가며 탐색을 한다.
5. start와 end를 더해 2로 나눈 중간값을 mid에 저장한다.
6. 이제 mid의 시간동안 심사위원들이 n명의 사람을 심사 할 수 있는지 확인해야 한다.
7. mid의 시간동안 심사위원들이 총 심사 할 수 있는 인원을 구하기 위해 long형 변수 sum을 만들고 times를 for문을 돌면서 mid를 times[i]로 나눈 값을 sum에 더해준다. (mid / times[i]는 총 mid의 시간동안 각 심사위원들이 검사 할 수 있는 인원수의 합이다.)
8. 따라서 sum이 n보다 같거나 크면 심사해야 할 인원인 n명을 심사했거나 더 많이 한 것이므로 answer에 mid를 저장하고 end를 mid - 1로 하여 시간을 줄여서 더 최선의 경우인 시간을 탐색한다.
9. sum이 n보다 작다면 심사해야 할 인원인 n명보다 작게 심사한 것이므로 start를 mid + 1로하여 시간을 늘려서 탐색한다. 
10. 이렇게 탐색범위를 반식 줄여가며 탐색하다가 start가 end보다 커지게 되면 탐색을 종료하고 n명을 심사할 수 있는 최선의 시간을 저장한 answer를 반환한다.

 

 

 

참고 : https://youngest-programming.tistory.com/m/499 , https://geunzrial.tistory.com/m/39