BoJ 10816 [python]

입력
첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 몇 개 가지고 있는 숫자 카드인지 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.

출력
첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 몇 개 가지고 있는지를 공백으로 구분해 출력한다.

실패한 코드(메모리 초과)
동일한 수의 갯수를 구하는 문제이기에 '인덱스 = 수'인 리스트를 만드는것이 좋을것이라 판단했지만 실패

import sys

n = int(sys.stdin.readline().rstrip())
arrn = list(map(int, sys.stdin.readline().rstrip().split()))

#크기 만큼 배열 생성. [0] 은 양수, [1]은 음수
if abs(max(arrn)) >= abs(min(arrn)):
  arr = [[0,0] for i in range(2+abs(max(arrn)))]
  
else: 
  arr = [[0,0] for i in range(2+abs(min(arrn)))]


for i in arrn:
  if i >= 0 :
    arr[i][0] += 1
  else:
    arr[-i][1] += 1

m = int(sys.stdin.readline().rstrip())
arrm = list(map(int, sys.stdin.readline().rstrip().split()))

for i in arrm:
  if i >=0:
    print(arr[i][0],end= ' ')
  else:
    print(arr[-i][1], end = ' ')

성공한 코드
dict를 이용하여 필요한 정보만 저장하여 메모리를 줄였습니다.

import sys

n = int(sys.stdin.readline().rstrip())
arrn = list(map(int, sys.stdin.readline().rstrip().split()))

dic = {}

for i in arrn:
  dic[i] = 0
for i in arrn:
  dic[i] +=1


m = int(sys.stdin.readline().rstrip())
arrm = list(map(int, sys.stdin.readline().rstrip().split()))

for i in arrm:
  try:
    print(dic[i],end = ' ')
  except KeyError:
    print(0, end = ' ')