[알고리즘] 백준 1920 - 수 찾기

문제

수 찾기

접근

1. 이코테의 부품 찾기와 동일하다

2. 이진 탐색을 통해서 res에 함수의 값을 덮고 그 값이 None이 아니라면, 1을 출력 맞다면, 0 을 출력한다

 

풀이

# 수 찾기 - 부품찾기랑 비슷
import sys
input = sys.stdin.readline

n = int(input())
array = list(map(int, input().split()))
m = int(input())
m_list = list(map(int, input().split()))

array.sort()


def binary_search(n_list, target, start, end):
    while start <= end:
        mid = (start+end)//2
        if n_list[mid] == target:
            return mid
        elif n_list[mid] > target:
            end = mid - 1
        elif n_list[mid] < target:
            start = mid + 1
    return None


for i in m_list:
    res = binary_search(array, i, 0, n-1)
    if res != None:
        print('1', end='\n')
    else:
        print('0', end='\n')