[1920] 수 찾기 (실버 4)

1920번: 수 찾기

n개의 수가 주어지고 m개의 수가 주어질 때, m개의 수가 n개의 수 안에 속하는지 출력하는 문제다.

풀이과정은 다음과 같다.

• n개의 수를 defaultdict(default 0 )에 1로 저장한다.
• m개의 수를 defaultdict에서 출력한다.

매번 n개의 수를 탐색하지 않는 방법을 떠올리는 쉬운 문제였다.

import sys
import collections

n = int(sys.stdin.readline().strip())
numbers = map(int, sys.stdin.readline().strip().split(' '))
numbersDict = collections.defaultdict(lambda : 0)
for number in numbers:
  numbersDict[number] = 1
m = map(int, sys.stdin.readline().strip().split(' '))
checkNumbers = map(int, sys.stdin.readline().strip().split(' '))
for checkNumber in checkNumbers:
  print(numbersDict[checkNumber])