[백준 10815] 숫자 카드 with Python

문제 링크

www.acmicpc.net/problem/10815

10815번: 숫자 카드

풀이

import sys

n = int(sys.stdin.readline())
n_lst = list(map(int, sys.stdin.readline().split()))

m = int(sys.stdin.readline())
m_lst = list(map(int, sys.stdin.readline().split()))

def binary_search(lst, n):
    low = 0
    high = len(lst)-1
    
    while low <= high:
        mid = (low+high)//2

        if n > lst[mid]:
            low = mid+1
        elif n < lst[mid]:
            high = mid-1
        else:
            return 1
    
    return 0

n_lst.sort()
answer = []
for i in m_lst:
    answer.append(binary_search(n_lst, i))
print(" ".join(map(str, answer)))

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