[백준] 1920 | 수 찾기 [파이썬/python]

문제

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

두 개의 자연수 목록을 받아 두 번째 목록의 숫자들이 첫 번째 목록에 존재하는지를 확인하는 문제이다.

 

접근

처음 입력받은 자연수 목록을 set에 저장한 후, for문을 통해 두 번째 자연수를 하나씩 조회하며 결과를 출력하도록 했다.

 

코드

import sys
input = sys.stdin.readline

N = int(input())
A = set(map(int, input().split()))
M = int(input())

for b in map(int, input().split()):
    if b in A:
        print(1)
    else:
        print(0)