공부하자!/알고리즘

백준 10816번 숫자 카드 2 파이썬

지우개원정대 2021. 6. 11. 01:40

숫자 카드 1이랑 똑같은 문제다! 하고 방심했는데 아니었음..

해쉬테이블 배워둬서 정말 다행이야!!

 

 

A 안의 요소 있으면 해쉬테이블 안에 +1 씩 하고 나중에 그걸 출력하면 된다~

마지막에 출력할 때는 없는 부분은 0으로 바꿔야 함!

 

# 숫자 카드 2

import sys

N = int(sys.stdin.readline())
A = list(map(int, sys.stdin.readline().split()))
M = int(sys.stdin.readline())
B = list(map(int, sys.stdin.readline().split()))

hashmap = {}

for i in A:
    if i in hashmap:
        hashmap[i] += 1
    else:
        hashmap[i] = 1

print(' '.join(str(hashmap[i]) if i in hashmap else '0' for i in B))