[python] 백준 10815번

안녕하세요 !

오늘은 백준 - 숫자 카드 문제를 가져왔습니다.

문제 설명

해결 방법

문제를 해결한 방법은 다음과 같습니다.

1. 입력을 받는다.
2. M개의 숫자를 key로 0을 value로 해서 딕셔너리를 생성한다.
3. N개의 숫자가 딕셔너리에 있으면 value 값을 1로 수정한다.
4. value 값을 수정한다.

1. 입력을 받는다.

card_num = int(sys.stdin.readline().strip())
card_arr = list(map(int, sys.stdin.readline().strip().split()))
number = int(sys.stdin.readline().strip())
num_arr = list(map(int, sys.stdin.readline().strip().split()))

2. M개의 숫자를 key로 0을 value로 해서 딕셔너리를 생성한다.

d = {}
for n in num_arr:
    d[n] = 0

3. N개의 숫자가 딕셔너리에 있으면 value 값을 1로 수정한다.

for c in card_arr:
    if c in d:
        d[c] = 1

4. value 값을 수정한다.

for key, value in d.items():
    print(value, end=" ")

전체 코드

import sys

card_num = int(sys.stdin.readline().strip())
card_arr = list(map(int, sys.stdin.readline().strip().split()))

number = int(sys.stdin.readline().strip())
num_arr = list(map(int, sys.stdin.readline().strip().split()))

d = {}
for n in num_arr:
    d[n] = 0

for c in card_arr:
    if c in d:
        d[c] = 1
        
for key, value in d.items():
    print(value, end=" ")