[백준][Python] 10816번 숫자 카드 2

10816번: 숫자 카드 2

 

코드

import sys
from collections import defaultdict

dic = defaultdict(int)

n = int(input())
n_list = list(map(int,sys.stdin.readline().split()))
for i in n_list:
	dic[i] += 1

m = int(input())
m_list = list(map(int,sys.stdin.readline().split()))
ans = []
for i in m_list:
	ans.append(dic[i])

print(*ans)

 

설명

defaultdict를 활용하여 모든 키값에 대한 value를 정수 0으로 초기화를 시켜놨다.

상근이가 가지고 있는 숫자 i 를 for문으로 하나씩 받아 dic[i] += 1 을 해서 개수를 세어준다.

그리고 m개의 수를 받아 딕셔너리에서 해당 키에 대한 값을 찾으면 개수를 알아낼 수 있다.