[백준][Python] 2776번 암기왕

2776번: 암기왕

 

코드

import sys
T = int(input())
for _ in range(T):
    N = int(input())
    num1 = set(sys.stdin.readline().split())
    M = int(input())
    num2 = list(sys.stdin.readline().split())
    for i in num2:
        if i in num1:
            print(1)
        else:
            print(0)

 

설명

수첩1의 숫자는 중복된 값은 필요없기 때문에 set으로 받아준다. list로 받게 되면 시간초과 오류가 뜬다.

수첩2는 각각의 값마다 0또는 1을 출력해야 하므로 list로 받아줬다.