[프로그래머스][이분탐색][C++] 입국심사

https://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명 이하입니다.

풀이 방법

이분 탐색은 정렬된 특정 데이터 집합에서 탐색하고자 하는 값을 기준으로 가능한 최솟값, 최댓값의 범위에서 중간값으로 범위를 조절해가며 탐색하는 방법이다.

이번 문제에서 찾고자하는 값은 모든 사람이 심사를 받는데 걸리는 시간이므로 심사를 받는 시간을 기준으로 잡는다.

심사의 최솟값은 1명이 1만큼 걸리는 심사를 받는 것이므로 1이고 최댓값은 times 벡터에서 가장 큰 값 * n이다.

그럼 범위를 조절해줄 조건을 찾아야 하는데, 모든 사람이 심사를 받아야 하므로 mid 시간에서 심사를 받을 수 있는 사람의 수인 cnt를 만들었다.

cnt >= n일 경우 해당 시간에서 모든 사람이 심사를 받을 수 있으므로 answer에 mid값을 넣어주고 mid값을 더 작게 해주기 위해 right = mid - 1을 해준다.

반대일 경우에는 모든 사람이 심사를 받을 수 없는 경우이므로 mid값을 더 크게 해주기 위해 left = mid + 1을 해주면서 문제를 해결했다.

소스 코드

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

using namespace std;

long long solution(int n, vector<int> times) {
    long long answer = 0;
    
    sort(times.begin(), times.end());
    
    long long left = 1, right = (long long) times[times.size() - 1] * (long long) n, mid, cnt = 0;
    
    while(left <= right) {
        mid = (left + right) / 2;
        
        for(auto i : times)
            cnt += mid / i;
        
        if(cnt >= n) {
            answer = mid;
            right = mid - 1;
        }
        else
            left = mid + 1;
        
        cnt = 0;
    }
    
    return answer;
}