10816번 - 숫자 카드 2

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

10816번: 숫자 카드 2

리스트로 하면 무조건 시간 초과가 날 것 같았다. 따라서 딕셔너리를 사용하여 해당 카드의 수가 몇 번 나왔는지 value로 저장하였다.

즉, card라는 list에서 10이 3번 나왔으면 {10 : 3} 이렇게 저장되는 것이다.

그리고 만약에 find_number에 일치하는 키가 있다면 값을 출력해주면 되고 만약 없다면 0을 출력한다.

딕셔너리에서 in 연산은 시간 복잡도가 O(1) 이기 때문에 빠르게 가능하다. 

# 숫자 카드 2
import sys
input = sys.stdin.readline
N = int(input())
card = list(map(int, input().split()))
M = int(input())
find_number = list(map(int, input().split()))
dict_card = {}
answer = []
for i in range(N):
    if dict_card.get(card[i]):
        dict_card[card[i]] += 1
    else:
        dict_card[card[i]] = 1
for key in find_number:
    if key in dict_card:
        answer.append(dict_card[key])
    else:
        answer.append(0)
    
print(*answer)