[python] 백준 1902번: 수 찾기

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

 

1920번: 수 찾기

 

 

 

코딩테스트용 언어로 C++ 만을 파고 있었는데, 이번에 파이썬으로 시작해보고자 한다.

 

코드

N = input()
A = map(int, input().split())
M = input()
B = map(int,input().split())
A = list(A)
B = list(B)

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

 

 

 

알게 된것 

A = map(int, input().split())

공백으로 구분하여 입력 받는다.

map은 리스트의 요소를 지정된 함수로 처리해주는 함수이다.

문제는 map 함수의 return object 는 map object 이기 때문에

list 로 변환해주지 않으면 원하는 결과가 나오지 않는다는 것이다.

 

위에는 지저분하게 list 변환을 아래에 썼지만

B = list(map(int,input().split()))

앞으로는 list로 wrapping 하도록 하자.