[파이썬, Python] 백준 10816: 숫자 카드 2

문제

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

10816번: 숫자 카드 2

코드

import sys
input = sys.stdin.readline

n = input()
card_list = list(map(int,input().split()))
m = input()
num_list = list(map(int,input().split()))

cnt = {}

for i in card_list:
    if i in cnt:
        cnt[i] += 1
    else:
        cnt[i] = 1
        
for i in num_list:
    if i in cnt:
        print(cnt[i], end = ' ')
    else:
        print(0, end = ' ')

설명

이 문제 또한 딕셔너리를 이용하면 해결이 가능하다.

카운트를 위한 딕셔너리를 하나 만들어주고,
만약 card_list에 있는 카드가 cnt에 존재하지 않는다면 1을 넣어주고,
이미 존재하고 있다면 그 수에 해당하는 값에 1을 추가해준다.

최종적으로 가지고 있는 카드는 개수를 출력해주고,
그렇지 않은 카드는 0으로 출력해준다.
이 때, 출력 형식을 맞추기 위해 (end = ' ')을 사용하여 공백으로 출력한다.