본문 바로가기

프로그래머스 고득점kit/이분탐색

[고득점kit]이분탐색 #1. 입국심사

728x90

(문제 설명)

n명이 입국심사를 위해 줄을 서서 기다리고 있습니다. 각 입국심사대에 있는 심사관마다 심사하는 데 걸리는 시간은 다릅니다.

처음에 모든 심사대는 비어있습니다. 한 심사대에서는 동시에 한 명만 심사를 할 수 있습니다. 가장 앞에 서 있는 사람은 비어 있는 심사대로 가서 심사를 받을 수 있습니다. 하지만 더 빨리 끝나는 심사대가 있으면 기다렸다가 그곳으로 가서 심사를 받을 수도 있습니다.

모든 사람이 심사를 받는데 걸리는 시간을 최소로 하고 싶습니다.

입국심사를 기다리는 사람 수 n, 각 심사관이 한 명을 심사하는 데 걸리는 시간이 담긴 배열 times가 매개변수로 주어질 때, 모든 사람이 심사를 받는 데 걸리는 시간의 최솟값을 return 하도록 solution 함수를 작성해주세요.

제한사항
  • 입국심사를 기다리는 사람은 1명 이상 1,000,000,000명 이하입니다.
  • 각 심사관이 한 명을 심사하는데 걸리는 시간은 1분 이상 1,000,000,000분 이하입니다.
  • 심사관은 1명 이상 100,000명 이하입니다.
입출력 예ntimesreturn
6 [7, 10] 28
입출력 예 설명

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

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

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

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

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


(문제 해설)

 

! 풀기전에 생각해보아요!

 

심사를 해야하는 사람 6명

심사의원들이 심사 대기 시간 각각 7, 10분

조금만 깊게 생각해보면 10분 걸리는 심사의원한테 6번 받으면 최대 심사 시간이 나온다.

이렇게 심사시간을 계산하는 것부터가 이 문제 풀이의 첫 실마리입니다.

다음으로, 이분탐색이라는 주제에 맞게 0 <- 말이 안되지만 가장 최소로 걸리는 시간, 아까 구한 맥스 시간을 

start와 max로 두고 이분탐색을 할 것입니다.

 

 

== 문제 풀이 알고리즘 ==

 

예제 1을 예시로 구현할 알고리즘을 설명합니다.

0. mid = (st + end) // 2 ==> 30  # mid 예상 답 후보가 됩니다.

1. mid만큼 걸린다고 가정합니다. 

2. 각각 심사의원들이 mid(30)만큼 심사를 한다고 가정하고 심사를 합니다

2-1. 30 // 7 = 4, 30 // 10 = 3 ==> 총 candi(명): 4 + 3 = 7 명으로 우리가 심사하게 될 n(6) 보다 많이 심사하게 됩니다.

3. 결론적으로, 현재 산정한 mid(분)은 너무 시간이 많은 것

4. end = mid - 1로 줄여서 다시 mid를 계산 ---> 1로 이동

5. 만약, 더한 candi(명)이 심사해야 할 n명보다 작다면 st = mid + 1로 키우는 방식으로 시간을 산정해서 다시 계산합니다 ---> 1로 이동

6. st > end 가회면 해당 알고리즘을 종료 -> mid 답 출력

 


[ 코드 ]

 

def solution(n, times):
    answer = 0
    st = 0
    end = max(times) * n
    
    while st <= end:
        res= 0
        mid = (st + end) // 2
        
        # 중요 부분
        for t in times:
            res += mid // t
        
        if res >= n:
            answer = mid
            
            end = mid - 1
        else:
            st = mid + 1
            
    return answer

후기

 

이분 탐색을 문제를 정말 빠르고 쉽게 풀 수 있다.

이 문제는 알면 쉽지만 모르면 난감할 수 있습니다

728x90