[백준/파이썬] 1920번 수 찾기 풀이

업데이트:



문제 정보


풀이

문제

집합 A와 질의 수열 B가 주어질 때, 각 B 원소가 A에 존재하는지 출력하는 문제입니다.

코드

_ = int(input())
a = set(map(int, input().split()))
_ = int(input())
b = list(map(int, input().split()))

for each_b in b:
    print(1 if each_b in a else 0)

설명

Aset으로 만들어 membership 조회를 O(1) 평균 시간에 처리합니다.

질의마다 포함 여부를 1/0으로 출력합니다.



댓글남기기