10815번 - 숫자 카드

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

10815번: 숫자 카드

# 숫자 카드
import sys
input = sys.stdin.readline
N = int(input())
card = set(list(map(int, input().split())))
M = int(input())
find = list(map(int, input().split()))
answer = []
for idx in range(len(find)):
    if find[idx] in card:
        answer.append(1)
    else:
        answer.append(0)
print(*answer)

시간 초과를 피하기 위해서는 set 자료형을 사용하면 된다.

list에서의 in은 시간 복잡도가 O(n)인 반면, set 자료형에서의 in은 시간 복잡도가 O(1)이기 때문에 훨씬 효율적이다.

문제에서 card는 중복되지 않는 수라고 했기 때문에 가능하다.