티스토리 뷰

PS/Python

[Python] 백준 1920번: 수 찾기

yoo.o 2020. 10. 15. 02:21
반응형

문제

 

1920번: 수 찾기

첫째 줄에 자연수 N(1≤N≤100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1≤M≤100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안��

www.acmicpc.net

풀이

시간을 줄이기 위해 

1. 들어오는 수들을 정수형으로 바꿔주지 않음

2. set 자료형을 사용했다.

어차피 같은 문자인지 비교만 하면 되기 때문에 스트링을 굳이 정수형으로 바꿔주지 않아 시간을 줄였고, set 자료형을 사용해 중복을 없앴다. 

import sys

N = sys.stdin.readline()
A = set(sys.stdin.readline().strip().split(" "))
M = sys.stdin.readline()
B = sys.stdin.readline().strip().split(" ")

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

반응형
저작자표시 비영리