[백준 1920] 수 찾기 with Python

문제 링크

www.acmicpc.net/problem/1920

1920번: 수 찾기

풀이

import sys

n = int(sys.stdin.readline())
arr = sorted(list(map(int, sys.stdin.readline().split())))
m = int(sys.stdin.readline())
arr2 = list(map(int, sys.stdin.readline().split()))

def search(num, arr):
    high = len(arr)-1
    low = 0
    while low <= high:
        mid = (high + low) // 2
        item = arr[mid]
        if item > num:
            high = mid - 1
        elif item < num:
            low = mid + 1
        else:
            return 1
    return 0

for i in arr2:
    print(search(i, arr))

이진 탐색을 이용하여 해결하였다.