[백준-2776] 암기왕 / Python

https://www.acmicpc.net/problem/2776

2776번: 암기왕

import sys

def binary_search(start, end, note1, num):
    while start <= end:
        mid = (start+end)//2
        if note1[mid] == num:
            return 1
        elif note1[mid] < num:
            start = mid + 1
        else:
            end = mid - 1
    return 0

for _ in range(int(sys.stdin.readline())):
    n = int(sys.stdin.readline())
    note1 = list(map(int, sys.stdin.readline().split()))
    m = int(sys.stdin.readline())
    note2 = list(map(int, sys.stdin.readline().split()))
    note1.sort()
    for num in note2:
        print(binary_search(0, n-1, note1, num))

 

해설

간단한 이분탐색 문제. 노트1을 정렬해준 후, 노트2의 데이터로 노트1에서 이분탐색을 진행하면 된다.