Algorithm
[백준 1920] 수 찾기 - 파이썬
choiht
2023. 4. 24. 16:03
반응형
문제
https://www.acmicpc.net/problem/1920
1920번: 수 찾기
첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들
www.acmicpc.net
풀이
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를 리스트로 받았다가 시간 초과 오류가 났다.
반응형