본문 바로가기

알고리즘/백준 (Pyhthon)

[알고리즘] 백준 1920 수찾기

n = int(input())
nn = set(map(int, input().split()))
m = int(input())
mm = list(map(int, input().split()))

for i in range(m):
    if mm[i] in nn:
        print('1')
    else:
        print('0')

 

더 최적화 시킨 코드

N, A = int(input()), {i:1 for i in map(int, input().split())}
M = int(input())

for i in list(map(int, input().split())):
    print(A.get(i, 0))
딕셔너리를 confrehension으로 만들기
for문으로 입력받기
get함수를 이용해 1 또는 0 출력