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

문제설명은 더보기를 눌러주세요

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

 

-풀이- 

1. 모든 사람이 심사를 받는데 걸리는 시간의 최솟값을 이진탐색을 이용하여 구하는 문제입니다.

2. 모든 사람이 심사를 받는데 걸리는 시간의 범위
    : 1 ~ 10^18 (심사관이 한 명을 검사하는 데 걸리는 시간의 최댓값(10^9) x 입국심사를 기다리는 사람 수의 최댓값(10^9))
    10^18 < 2^63-1 이므로, long long 보다 작습니다. 

    그런데 저는 굳이 10^18로 안하고,
    심사관이 한 명을 검사하는 데 걸리는 시간 중 최댓값 x 입국심사를 기다리는 사람 수를 최댓값으로 설정했습니다.

3. 모든 사람이 심사를 받는데 걸리는 시간의 범위를 이진탐색하며, 조건을 만족하는 최솟값을 구하면 됩니다.
    조건이란, 해당 시간 내에 모든 사람들의 심사가 끝나는지 여부를 말합니다.

4. 해당 시간 내에 모든 사람들의 심사가 끝나는지 여부를 파악하려면,   
    (시간 / 심사관이 한 명을 심사하는데 걸리는 시간)을 통해,
    해당 시간 동안 심사관이 심사할 수 있는 사람 수를 구할 수 있습니다.
    모든 심사관에 대해 심사할 수 있는 사람수를 더하면, 해당 시간 동안 모든 심사관이 심사할 수 있는 사람 수가 됩니다.

5. 해당 시간동안 모든 심사관이 심사할 수 있는 사람 수가
    n보다 작다면 해당 시간을 더 키워야겠고, n보다 크거나 같다면 해당 시간을 더 줄여야 합니다. 
    이렇게 n보다 크거나 같은 경우 중 최솟값을 구하면 모든 사람이 심사를 받는데 걸리는 시간의 최솟값을 구할 수 있습니다.
    

-코드(C++)-

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>
typedef long long ll;
using namespace std;

ll solution(int n, vector<int> times) {
    ll answer = 0, min_time = 1, mid_time = 0;
    ll max_time = *max_element(times.begin(), times.end()) * (ll)n;
    while(min_time <= max_time){
        mid_time = (min_time + max_time)/2;
        ll sumTime = 0;
        for(int i=0; i<times.size(); i++){
            sumTime += (mid_time/times[i]);
        }
        if(sumTime >= n) {
            answer = mid_time;
            max_time = mid_time-1;
        } else {
            min_time = mid_time+1;
        }
    }
    return answer;
}

-코드(Swift)-

import Foundation
func solution(_ n:Int, _ times:[Int]) -> Int64 {
    var res : Int64 = 0
    var st : Int64 = 1
    var en = Int64(times.max()!) * Int64(n)
    while st <= en {
        let mid = (st + en) / 2
        var sum_people : Int64 = 0
        for i in times {
            sum_people += mid/Int64(i)
        }
        if(sum_people >= n) {
            res = mid
            en = mid - 1
        } else {
            st = mid + 1
        }
    }
    return res
}