백준 1920 수 찾기 파이썬

[문제 링크]

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

[입출력 예]

입력	출력
5 4 1 5 2 3 5 1 3 7 9 5	1 1 0 0 1

[소스코드]

import sys
input=sys.stdin.readline
from collections import deque

input()
a=list(map(int,input().split()))
input()
test=list(map(int,input().split()))
for i,x in enumerate(test):
    test[i]=[x,i,0]

a.sort()
test.sort()
a=deque(a)
test=deque(test)
ans=[]

while(a and test):
    cur_a=a.popleft()
    cur_test=test.popleft()

    if(cur_a<cur_test[0]):
        test.appendleft(cur_test)
        continue
    elif(cur_a>cur_test[0]):
        a.appendleft(cur_a)
        ans.append(cur_test)
        continue
    else:
        cur_test[2]=1
        ans.append(cur_test)
        a.appendleft(cur_a)
while(test):
    ans.append(test.popleft())


ans=sorted(ans,key=lambda x:x[1])
print('\n'.join(list(map(lambda x:str(x[2]),ans))))

[소스코드2]

import sys
input=sys.stdin.readline

input()
a=set(input().split())
ans=''
input()
for i in input().split():
    if i in a:
        ans+='1\n'
    else:
        ans+='0\n'
print(ans)

# set 을 사용하지 않으면 시간 초과 에러가 난다.