[프로그래머스] 이분탐색 - 입국심사 (Java)

프로그래머스 - 이분탐색 - 입국심사

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

문제 설명

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분에 모든 사람의 심사가 끝납니다.

public static long solution(int n, int[] times) {
    long answer = 0;
    int len = times.length;
    Arrays.sort(times);
    long l = 0;
    // r = 최대소요시간 * 총 인원 수
    long r = (long)times[len - 1] * n;
    while (l <= r) {
        long mid = (l + r) / 2;
        long sum = 0;
        for (int time : times) {
            // 합계 = 중간 값 / 시간 (중간값에서 각 시간요소 마다 몇명이 심사 받는지)
            sum += mid / time;
        }
        if (sum >= n) {
            // 계산된 총 인원수가 주어진 인원 n 보다 크다면 sum을 감소시켜야 함
            r = mid - 1;
            answer = mid;
        } else {
            l = mid + 1;
        }
    }
    return answer;
}

처음에는 어떻게 풀어야할지 막막했지만, 확실한 기준을 잡고 시작하니까 풀렸다.

• 탐색할 범위 : 시간 배열 (times)
  • 맨 왼쪽(l) : 0
  • 맨 오른쪽(r) : 최대 입국심사 시간 * 총 인원 수 (총 입국심사의 최댓값)
• target 원소: 총 인원 (n)

이분탐색에서 첫번째로 해야할 일은 수열을 정렬해주는 것이다.

그다음에는 수열의 범위를 설정해주는 것이다 (l, r)

타겟인 인원수를 구하기 위해서는 범위에 해당하는 원소(mid)를 각 times의 원소들로 나누어서 변수 sum에 더한다. 그러면 mid 시간동안 입국심사 할 수 있는 인원이 나온다.

sum이 n보다 큰 경우는 r을 감소시켜 sum의 값을 줄인다. n보다 작은 경우에는 l 을 증가시켜 sum의 값을 증가시킨다.

l과 r이 같아질 때 까지 반복해준다.