[프로그래머스 / 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

 

 

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

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

 

 

 

 

문제 풀이

 

이분 탐색을 이용해 문제를 해결합니다. 모든 사람을 심사하는 시간의 min은 1, max는 제일 오래 걸리는 심사관이 모든 사람을 처리할 때 걸리는 시간입니다. 따라서 max는 (times 중 가장 큰 원소 * n)이 됩니다. 

 

어떤 시간이 주어졌을 때, 주어진 시간에 입국심사할 수 있는 모든 사람의 수를 합쳐 people_sum에 저장합니다.

 

이 people_sum이 n보다 작다면 정답은 mid보다 크므로 min = mid + 1; 을 합니다. 반대로 people_sum이 n이상이라면 정답이 mid보다 작거나 같으므로 max = mid - 1; answer = mid; 를 합니다.

 

min이 max보다 커질 때까지 위의 과정을 반복합니다. 

 

 

 

 

 

 

소스코드

 

// 입국심사
#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
long long solution(int n, vector<int> times) {
    long long answer, people_sum;
    long long min = 1, max, mid;
    max = *max_element(times.begin(), times.end()) * (long long)n;
    answer = max;
 
    while (min <= max) {
        people_sum = 0;
        mid = (min + max) / 2;
 
        for (int i = 0; i < times.size(); i++)
            people_sum += mid / times[i];
 
        if (people_sum < n) {
            min = mid + 1;
        }
        else {
            max = mid - 1;
            answer = mid;
        }
    }
    
    return answer;
}

 

정확성 테스트

 

 

 

 

공부한 것을 정리한 내용입니다. 수정할 부분이 있다면 알려주시면 감사하겠습니다 :)