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

1. 문제 설명

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

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

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

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

 

---

2. 제한사항

• 입국심사를 기다리는 사람은 1명 이상 1,000,000,000명 이하입니다.
• 각 심사관이 한 명을 심사하는데 걸리는 시간은 1분 이상 1,000,000,000분 이하입니다.
• 심사관은 1명 이상 100,000명 이하입니다.

---

3. 입출력 예

 

n	times	return
6	[7, 10]	28

 

---

4. 나의 풀이

이진 탐색을 대체 어디에서 이용해야할지 모르겠어서 고민하는데 시간이 좀 오래 걸렸다. 우선 걸리는 시간이 될 수 있는 범위를 잡고, 답이 될 수 있는 시간을 이진 탐색 하며 찾으면 된다.

 

이분 탐색의 대상: 답이 될 수 있는 검색 시간의 범위

 

def solution(n, times):
    right = min(times) * n
    left = 1
    
    answer = 0
    while(left <= right):
        mid = (right + left) // 2
        temp = n
        for i in times:
            temp -= mid//i
            if temp <= 0:
                answer = mid
                right = mid - 1
                break
        if temp>0:
            left = mid + 1
                
    return answer

 

숫자의 범위는 left = 1 부터 right = 가장 시간이 적게 걸리는 심사대에서 모든 사람이 심사받는 경우의 시간을 넣어 주었다.

중앙값을 찾고, 심사대 리스트를 순회하며 각각의 심사대에서 수용할 수 있는 사람의 수를 줄여가며 심사대를 모두 이용하지 않았는데 모든 사람이 심사를 받는 경우에는 범위의 왼쪽 (전체의 반을 자른 기준으로 왼쪽), 모든 심사대를 순회했음에도 기다리는 사람이 남은 경우에는 범위의 오른쪽으로 줄여주고 다시 탐색한다.

이분 탐색 유형의 문제의 경우는 어떤 범위를 탐색하고 목표값이 무엇인지를 파악하는게 관건이라고 생각했다.

 

 

문제 링크: https://programmers.co.kr/learn/courses/30/lessons/43238?language=python3

코딩테스트 연습 - 입국심사