[BOJ] 2776 암기왕

import sys

t=int(sys.stdin.readline())
for _ in range(t):
    n=int(sys.stdin.readline())
    note1=set(map(int, sys.stdin.readline().split()))
    m=int(sys.stdin.readline())
    note2=list(map(int, sys.stdin.readline().split()))
    for i in note2:
        if  i in note1:
            print(1)
        else:
            print(0)

<코드 설명>

testcase t를 먼저 입력받아주었다.

t만큼 for문으로 반복시켜 수첩1의 정수 개수인 n, n개의 수첩1 정수들, 수첩2의 정수 개수인 m, m개의 수첩2 정수들을 모두 입력받아주었다.

note2 속 원소인 i가 note1에 속해 있다면 1을 출력해주고, 그렇지 않으면 0을 출력하도록 하였다.

시간초과를 대비하여 import sys를 이용하여 입력받아주었다.

 

 

<실행결과>