[백준/파이썬] 1920번 수 찾기 풀이
업데이트:
문제 정보
- 문제 출처: 백준 온라인 저지
- 문제 링크: 1920번 수 찾기
- 문제풀이 코드 GitHub 링크
- 제출 언어: Python 3
풀이
문제
집합 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)
설명
A를 set으로 만들어 membership 조회를 O(1) 평균 시간에 처리합니다.
질의마다 포함 여부를 1/0으로 출력합니다.
댓글남기기