[10815] 숫자 카드 (Python)

[문제]

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

10815번: 숫자 카드

---

알고리즘 분류는 정렬, 이분 탐색 입니다.

2번째 줄에 입력받은 카드를 정렬하고, 4번째 줄에 입력받은 카드를 이분 탐색으로 찾아야 합니다.

---

[코드]

import sys


def binary(data, find):
    left = 0
    right = len(data) - 1

    while left <= right:
        mid = (left + right) // 2

        if data[mid] == find:
            return "1"
        elif data[mid] < find:
            left = mid + 1
        elif data[mid] > find:
            right = mid - 1

    return "0"


N = int(sys.stdin.readline())
card_N = sorted(list(map(int, sys.stdin.readline().split())))
M = int(sys.stdin.readline())
card_M = list(map(int, sys.stdin.readline().split()))

res = list()
for i in card_M:
    res.append(binary(card_N, i))

for i in res:
    print(i, end=" ")