[프로그래머스] 입국심사 / C++

문제 설명

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

 

 

코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

long long solution(int n, vector<int> times) {
    long long answer;
    long long minTime, maxTime, avgTime;
    long long man;
    int i;
    
    minTime = 1;
    maxTime = *max_element(times.begin(), times.end()) * (long long)n;
    while (minTime <= maxTime)
    {
        man = 0;
        avgTime = (minTime + maxTime) / 2;
        for (i = 0; i < times.size(); i++)
        {
            man += avgTime / times[i];
        }
        if (man < n)
        {
            minTime = avgTime + 1;
        }
        else
        {
            maxTime = avgTime - 1;
            answer = avgTime;
        }
    }
    return answer;
}

 

입력값의 범위는 1 <= n, times[i] <= 1,000,000,000이다. 따라서 결과값이 최소가 되는 경우는 1명을 1분 동안 심사하는 경우이다. 또한, 결과값은 times의 원소 중 최댓값과 n을 곱한 값보다 작거나 같다.

이분 탐색의 초기 범위를 1 ~ max_element(times) * n으로 설정한다. 두 값의 평균이 중간값인 avgTime이 된다. avgTime에 이르기까지 검사 가능한 인원이 n보다 적으면 시간이 더 필요한 것이므로 minTime을 갱신해서 avgTime을 늘리고, n보다 크거나 같으면 n명을 검사하기에 충분한 시간인 것이므로 maxTime을 갱신해서 avgTime을 줄인다. n명을 심사하기 위해 필요한 최소 시간을 리턴하는 것이므로 avgTime에 검사 가능한 인원이 n명 이상일 때, answer를 avgTime으로 갱신한다.

 

링크

programmers.co.kr/learn/courses/30/lessons/43238

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

github.com/inbdni/Programmers/blob/master/level03/%EC%9E%85%EA%B5%AD%EC%8B%AC%EC%82%AC.cpp

inbdni/Programmers