이것이 코딩테스트다 with 파이썬 - Chp 7. 이진 탐색_2. 부품 찾기

Try1

n = int(input())
array1 = list(map(int, input().split()))
m = int(input())
array2 = list(map(int, input().split()))

for i in range(len(array2)):
    if array2[i] in array1:
        print("yes")
    else:
        print("no")

no
yes
yes
-> 답안은 한줄에 나오도록 해야한다.(조건)

Try2

n = int(input())
array1 = list(map(int, input().split()))
m = int(input())
array2 = list(map(int, input().split()))

for i in m:
    if array2[i] in array1:
        print("yes", end=" ")
    else:
        print("no", end=" ")

no yes yes
-> 이진 탐색을 사용하여 문제를 풀어야 했다.

Try3

n = int(input())
array1 = list(map(int, input().split()))
m = int(input())
array2 = list(map(int, input().split()))

array1.sort()

for i in array2:
    start = 0
    end = n
    while start <= end:
        mid1 = (start + end) // 2
        if array1[mid1] > i:
            end = mid1 - 1
        elif array1[mid1] < i:
            start = mid1 + 1
        elif array1[mid1] == i:
            print('yes', end=' ')
            break
    else:
        print('no', end=' ')

논리구조는 맞는 것 같았고, 흐름도 이해가 되는데 답이 계속 도출되지 않아서 시행착오를 많이 겪었다.

Try4

n = int(input())
array1 = list(map(int, input().split()))
m = int(input())
array2 = list(map(int, input().split()))

array1.sort()

def binary_search(array1, i, end, start):
    mid1 = (start + end) // 2
    if start <= end:
        if array1[mid1] > i:
            return binary_search(array1, i, mid1 - 1, start)
        elif array1[mid1] < i:
            return binary_search(array1, i, end, mid1 + 1)
        elif array1[mid1] == i:
            return print("yes", end=' ')
    else:
        return print("no", end=' ')


for i in array2:
    binary_search(array1, i, end=n, start=0)

재귀함수로 한번 코드를 구성해보았다.

예시 답안

# 이진 탐색 소스코드
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=' ')