코딩테스트/프로그래머스

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

멍토 2019. 12. 16.

※ 저의 풀이가 무조건적인 정답은 아닙니다.

다른 코드가 좀더 효율적이고 좋을 수 있습니다.

다른사람들의 풀이는 언제나 참고만 하시기 바랍니다.

문제 주소입니다.

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

 

코딩테스트 연습 - 입국심사 | 프로그래머스

n명이 입국심사를 위해 줄을 서서 기다리고 있습니다. 각 입국심사대에 있는 심사관마다 심사하는데 걸리는 시간은 다릅니다. 처음에 모든 심사대는 비어있습니다. 한 심사대에서는 동시에 한 명만 심사를 할 수 있습니다. 가장 앞에 서 있는 사람은 비어 있는 심사대로 가서 심사를 받을 수 있습니다. 하지만 더 빨리 끝나는 심사대가 있으면 기다렸다가 그곳으로 가서 심사를 받을 수도 있습니다. 모든 사람이 심사를 받는데 걸리는 시간을 최소로 하고 싶습니다. 입국심사

programmers.co.kr


1. 문제 설명

2. 문제 해석

3. 소스 코드

3.1 주석 없는 코드

3.2 주석 있는 코드

3.3 테스트 코드

4. 결과


1. 문제 설명

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
1 {2, 2} 2
1 {1} 1

 

입출력 예 설명

가장 첫 두 사람은 바로 심사를 받으러 갑니다.

7분이 되었을 때, 첫 번째 심사대가 비고 3번째 사람이 심사를 받습니다.

10분이 되었을 때, 두 번째 심사대가 비고 4번째 사람이 심사를 받습니다.

14분이 되었을 때, 첫 번째 심사대가 비고 5번째 사람이 심사를 받습니다.

20분이 되었을 때, 두 번째 심사대가 비지만 6번째 사람이 그곳에서 심사를 받지 않고 1분을 더 기다린 후에 첫 번째 심사대에서 심사를 받으면 28분에 모든 사람의 심사가 끝납니다.

 


2. 문제풀이

이번에는 걸리는 시간을 기준으로 이분탐색을 했습니다.

최소시간을 1분, 최대시간을 인원수(n) * 심사하는데 제일 오래걸리는 시간으로 시작을 했습니다.

 

최소시간과 최대시간의 평균(avg)을 구합니다.

이 평균시간동안 각각 심사관들이 몇 명을 심사할 수 있는지 구해서 더합니다(human).

 n보다 평균시간동안 심사한 인원이 크거나 같다면(<=)

최대인원보다 많이 심사한것이므로 최대시간값을 평균-1로 바꿔주고

answer를 평균으로 갱신시킨다음에 다시 계산합니다.

 

심사한 인원이 n보다 적다면 최소시간값을 평균 +1 바꿔주고 다시 계산합니다.

한번 반복할때마다 심사한 사람(human)을 초기화합니다.

위 과정은 min값이 max값보다 클거나 같을때까지 반복합니다.

 

반복문을 탈출하게 된다면 answer에 최소시간이 담기게됩니다. 

 


3. 소스코드

3.1 주석없는 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#include <vector>
#include <algorithm>
 
using namespace std;
 
long long solution(int n, vector<int> times) {
    long long answer = 0, minTime = 1, maxTime, avgTime, human = 0;
    maxTime = *max_element(times.begin(), times.end()) * (long long)n;
    while (minTime <= maxTime) {
        avgTime = ((maxTime + minTime) / 2);
        for (auto t : times)        human += avgTime / t;
        if (n <= human) {
            answer = avgTime;
            maxTime = avgTime - 1;
        }
        else    minTime = avgTime + 1;
        human = 0;
    }
    return answer;
}

3.2 주석있는 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#include <vector>
#include <algorithm>
 
using namespace std;
 
long long solution(int n, vector<int> times) {
    //정답, 최소시간, 최대시간, 평균시간
    long long answer = 0, minTime = 1, maxTime, avgTime, human = 0;
    //최대값 은 최대인원수 * 인원수
    maxTime = *max_element(times.begin(), times.end()) * (long long)n;
    //최소시간이 최대시간과 같거나 적을때까지 반복
    while (minTime <= maxTime) {
        //평균시간 구하기
        avgTime = ((maxTime + minTime) / 2);
        //현재 시간으로 돌릴수 잇는 최대인원 구하기
        for (auto t : times)        human += avgTime / t;
        //최대인원이 총인원보다 크거나 같다면
        if (n <= human) {
            //현재 시간을 정답에 저장(나중에 안나오면 리턴해야 해서
            answer = avgTime;
            //최대시간 갱신
            maxTime = avgTime - 1;
        }
        //총인원보다 적다면 현재보다 시간이커야하므로 최소시간 갱신
        else    minTime = avgTime + 1;
        //사람수 초기화
        human = 0;
    }
    return answer;
}

3.3 테스트 코드 추가

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
 
using namespace std;
 
long long solution(int n, vector<int> times) {
    //정답, 최소시간, 최대시간, 평균시간
    long long answer = 0, minTime = 1, maxTime, avgTime, human = 0;
    //최대값 은 최대인원수 * 인원수
    maxTime = *max_element(times.begin(), times.end()) * (long long)n;
    //최소시간이 최대시간과 같거나 적을때까지 반복
    while (minTime <= maxTime) {
        //평균시간 구하기
        avgTime = ((maxTime + minTime) / 2);
        //현재 시간으로 돌릴수 잇는 최대인원 구하기
        for (auto t : times)        human += avgTime / t;
        //최대인원이 총인원보다 크거나 같다면
        if (n <= human) {
            //현재 시간을 정답에 저장(나중에 안나오면 리턴해야 해서
            answer = avgTime;
            //최대시간 갱신
            maxTime = avgTime - 1;
        }
        //총인원보다 적다면 현재보다 시간이커야하므로 최소시간 갱신
        else    minTime = avgTime + 1;
        //사람수 초기화
        human = 0;
    }
    return answer;
}
 
void print(int n, vector<int> times, long long answer){
    long long t = solution(n, times);
    if (t == answer)        cout << "정답" << endl;
    else        cout << "틀림" << endl;
}
 
int main(){
    print(6, { 710 }, 28);
    print(1, { 22 }, 2);
    print(1, { 1 }, 1);
    return 0;
}

4. 결과

 

댓글

💲 광고입니다.