[Python] 백준 #2776- 암기왕

문제

---

2776번: 암기왕

코드

---

My answer

import sys
input=sys.stdin.readline

t=int(input())

for i in range(t):
    n1=int(input())
    note1=list(map(int,input().split()))
    n2=int(input())
    note2=list(map(int,input().split()))
    result=set(note1)&set(note2)
    for j in note2:
        if(j in result):print(1)
        else:print(0)

Another answer

for c in range(int(input())):
    input()
    l1=set(map(int,input().split()))
    input()
    l2=list(map(int,input().split()))
    for i in l2:
        if i in l1:print(1)
        else :print(0)

풀이

---

그냥 교집합을 이용해서 풀었다. 주의할 점은 집합은 순서가 없기 때문에 수첩2에 적힌 숫자들을 집합자료형으로 선언하면 마지막에 순서대로 출력할 수가 없는걸 조심해라