[백준 1920] 수 찾기 - 파이썬

문제

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

1920번: 수 찾기

 

풀이

n = int(input())
a = set((map(int, input().split())))
m = int(input())
numbers = list((map(int, input().split())))

for i in numbers:
    print(1) if i in a else print(0)

 

1. a를 리스트가 아닌 set으로 받는다. (자동 중복 제거)

2. 입력받은 수를 numbers 라는 리스트에 넣는다.

3. numbers 내에 있는 수가 a 안에 존재하면 1을, 아니면 0을 출력한다. 

 

 

처음에는 a를 리스트로 받았다가 시간 초과 오류가 났다.