[이진 탐색 알고리즘] 부품 찾기 - 파이썬(python)

부품 찾기

난이도 : 中下 풀이 시간 : 30분

시간 제한 : 1초 메모리 제한 : 128 MB

 

---

 

해답

 

n = int(input())
goods = list(map(int, input().split()))
m = int(input())
wants = list(map(int, input().split()))

goods.sort()

def binary_search(array, target, start, end):
    while start <= end:
        mid = (start + end) // 2

        if array[mid] == target:
            return mid

        elif array[mid] > target:
            end = mid - 1

        elif array[mid] < target:
            start = mid + 1

    return None

result = 0

for i in wants:
    result = binary_search(goods, i, 0, n-1)

    if result == None:
        print('no', end= ' ')

    else:
        print('yes', end= ' ')

---

예시

 

# 이진 탐색 소스코드 구현 (반복문)
def binary_search(array, target, start, end):
    while start <= end:
        mid = (start + end) // 2
        # 찾은 경우 중간점 인덱스 반환
        if array[mid] == target:
            return mid
        # 중간점의 값보다 찾고자 하는 값이 작은 경우 왼쪽 확인
        elif array[mid] > target:
            end = mid - 1
        # 중간점의 값보다 찾고자 하는 값이 작은 경우 오른쪽 확인
        else:
            start = mid + 1
    return None

# N(가게의 부품 개수) 입력
n = int(input())
# 가게에 있는 전체 부품 번호를 공백을 기준으로 구분하여 입력
array = list(map(int, input().split()))
array.sort() # 이진 탐색을 수행하기 위해 사전에 정렬 수행
# M(손님이 확인 요청한 부품 개수) 입력
m = int(input())
# 손님이 확인 요청한 전체 부품 번호를 공백을 기준으로 구분하여 입력
x = list(map(int, input().split()))

# 손님이 확인 요청한 부품 번호를 하나씩 확인
for i in x:
    # 해당 부품이 존재하는지 확인
    result = binary_search(array, i, 0, n - 1)
    if result != None:
        print('yes', end=' ')
    else:
        print('no', end=' ')

---

해설

 

먼저 n, goods, m, wants 라는 입력을 받습니다. 이 문제는 이진 탐색 문제이므로 goods를 오름차순으로 정렬해줍니다.
goods가 이진 탐색 함수의 array가 되고, m의 각 입력값들이 이진 탐색 함수의 target 값이 될 것입니다.
입력을 받고 이진 탐색 함수를 정의해줍니다. 중간값이 target과 일치하면 그 인덱스를 반환합니다.
중간값이 target보다 크다면 end값을 중간값-1로, 중간값이 target보다 작다면 start를 중간값 +1로 설정해줍니다.
target 값을 찾을 수 없다면 None을 반환하고, 찾았다면 mid의 인덱스를 반환합니다.
이와 같이 이진 탐색 함수를 구현하고, 반복문을 통하여 wants의 각 값들을 target 값으로 설정하여 반복문을 돌립니다.