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

문제 설명

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

---

처음코드 :

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

using namespace std;

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

 

다른 사람의 풀이를 보고 보완한 코드 : 

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

using namespace std;

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

 

 

이분탐색에서 어떻게 입국심사를 받을 수 있는 사람을 구할 수 있을지 고민이 되었다.

그래서 다른 블로그를 참고하여 코드를 얻을 수 있었다.

 

코드 방식대로 예제를 풀이를 해보면 아래와 같다. 풀이는 보완된 코드를 기준으로 풀이하겠다.

 

해당 코드는 예상소요시간을 이분탐색을 통해 구하는 코드이다.

이분 탐색을 하는 방법은 정렬된 배열에서 최소값과 최대값의 가운데값과 구하려는 값을 비교하여 

만약 가운데 값이 구하려는 값보다 크다면 왼쪽을 작다면 오른쪽 부분을 탐색하는 방법이다.

 

이를 해당 문제에 대입해본다면 초기 배열은 다음과 같다.

첫 mid 값의 person 값은 4+3 = 7로 6보다 크므로 해당 코드를 지나게 된다.

 

 if (person >= n) {
            end = mid -1;
            if(answer >= mid)
                answer = mid;
        }

 

end값은 32가 되고 answer는 33이 저장된다.

다음 mid 값인 19의 person 값은 2+1 = 3으로 6보다는 작으므로 else 코드를 지나게 된다.

다음 mid 값은 26이고 person 값은 3+2 = 5로 6보다는 작고 else를 지난다.

다음 mid 값은 29이고 person 값은 4+2 = 6으로 if문을 지난다. answer은 29가된다.

다음 mid 값은 27이고 person 값은 3+2 = 5로 else 문을 지난다.

마지막 mid는 28이고 person 값은 4+2 = 6으로 answer는 28이 되고 while문을 빠져나가게된다.

그래서 답은 28이 된다.