[Python] 암기왕 - 백준

해당 글은 백준 2776번 문제 '암기왕'을 다룬다.

문제

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

2776번: 암기왕

 

설명

해당 문제는 의외로 간단하게 풀 수 있는 문제이다.

arr1에는 여러본 숫자도 하나만 저장되면 되므로 list가 아닌 set으로 입력받으면 된다.

처음에 list로 입력받았을 때, 시간초과 에러가 떠서 이분탐색으로 풀어야하나 했지만 중복만 제거하여 간단하게 풀 수 있었다.

 

코드

T = int(input())
for t in range(T):
    N = int(input())
    arr1 = set(map(int, input().split()))
    M = int(input())
    arr2 = list(map(int, input().split()))

    for i in arr2:
        if i in arr1:
            print(1)
        else:
            print(0)