알고리즘

<백준> 1920번 파이썬 알고리즘

changha. 2021. 7. 6. 15:26
import sys

a = int(sys.stdin.readline())
A = set(map(int, sys.stdin.readline().split()))
m = int(sys.stdin.readline())
M = list(map(int, sys.stdin.readline().split()))

for i in M:
    if i in A:
        print(1)
    else:
        print(0)

시간초과가 나서 구글링을 해보니

A부분을 set으로 바꿔줘야 했다.

list 는 O(n)이고 set은 O(1)라는 차이점 때문이다