Coding Test Practice/Python

[백준 - Silver 4] 1920번: 수 찾기

y2r1m 2023. 5. 9. 01:27

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())
B = list(map(int, input().split()))

for b in B:
    if b in A:
        print(1)
    else:
        print(0)

실버 4 문제 치고 매우 쉬운 편이다.

A를 입력 받을 때부터 set으로 저장해 시간 초과를 방지할 수 있다.

A를 리스트로 저장해두고 반복문 안에서 set(A)를 사용하면 시간이 초과된다.