🔏

[프로그래머스] 이진탐색 - 입국심사 / Python

LEVEL 3 - ‘입국심사’ 문제 보러 가기 !

문제 설명

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

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

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

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

문제 접근 방식

문제를 처음 보았을 때는 for 문을 돌면서 일일히 확인하는 문제인가 싶었지만 수의 범위가 너무 크기에 불가능해서 해당 방법으로는 풀 수 없다고 판단하여 이분 탐색의 개념에 대해 더 알아보고 문제에 접근했습니다.

문제의 핵심최소, 최대 범위를 구한 뒤 구하려고 하는 답을 이분 탐색으로 범위를 좁혀가며 답을 구하는 것 입니다.

  • 문제에서 최대 범위는 심사관 중 가장 오래걸리는 심사관을 계속 검사 받는 경우입니다.
  • 최소, 최대 범위의 중간값인 midn명을 심사 할 수 있는 지 아닌 지를 파악하며 이분 탐색을 진행합니다.

    • n명을 심사 할 수 있다면, 답을 갱신하고, 최대 범위를 줄여봅니다.
    • n명을 심사 할 수 없다면, 최소범위를 늘려봅니다.

풀이 코드

Python
def solution(n, times):
    answer = 0

    leng = len(times)
    left = 1
    right = (leng+1) * max(times) # 최대 범위

    while left <= right:
        mid = (left + right) // 2

        count = 0
        for time in times:
            count += mid // time
            # 심사 인원수를 넘으면 다음 단계
            if count >= n: break

        # n명을 심사 할 수 있는 경우
        # 한 심사관에게 주어진 시간을 줄여본다.
        if count >= n:
            answer = mid
            right = mid - 1
        # 없는 경우
        elif count < n:
            left = mid + 1

    return answer