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

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

10816번: 숫자 카드 2

# 백준 10816
import sys
input = sys.stdin.readline
N = int(input())
lst = list(map(int, input().split()))
field = [0]*(20000001)
for value in lst:
    field[value+10000000] += 1
M = int(input())
search = list(map(int, input().split()))
for value in search:
    print(field[value+10000000], end=' ')

단순한 배열을 이용하여 문제를 해결하였다.

 

from sys import stdin
_ = stdin.readline()
N = map(int,stdin.readline().split())
_ = stdin.readline()
M = map(int,stdin.readline().split())
hashmap = {}
for n in N:
    if n in hashmap:
        hashmap[n] += 1
    else:
        hashmap[n] = 1

print(' '.join(str(hashmap[m]) if m in hashmap else '0' for m in M))

딕셔너리를 이용하여 문제를 푸는 방법도 존재한다.