[파이썬, Python] 백준 2776: 암기왕

문제

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

2776번: 암기왕

코드

import sys
input = sys.stdin.readline

t = int(input())

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

설명

처음에 두 노트를 모두 리스트로 받았다가 시간초과가 나서 고민했던 문제였다.

1번 노트는 숫자가 중복으로 올 이유가 없기 때문에 set으로 받아준다.
2번 노트는 모든 숫자가 필요하므로 list로 받아줘야 한다.

그 이후 숫자를 비교해서 출력해 준다.