[Baekjoon/Python] 10816번: 숫자 카드 2 - 효과는 굉장했다!

10816번: 숫자 카드 2

알고리즘 분류

• 자료 구조
• 정렬
• 이분 탐색
• 해시를 사용한 집합과 맵

 

SOLUTION

해시를 이용한 풀이

import sys

N = int(sys.stdin.readline())
card = list(map(int, sys.stdin.readline().split()))
M = int(sys.stdin.readline())
card_cnt = list(map(int, sys.stdin.readline().split()))

hashmap = {} # 해시를 이용한 풀이
for n in card:
    if n in hashmap: # 이미 카운트 된 수일 경우
        hashmap[n] += 1

    else: # 처음 카운트 될 경우
        hashmap[n] = 1

for m in card_cnt:
    if m in hashmap: 
        print(hashmap[m], end = " ")
    
    else:
        print(0, end = " ")

 

카운터 함수를 이용한 풀이

from collections import Counter
import sys

N = int(sys.stdin.readline())
card = list(map(int, sys.stdin.readline().split()))
M= int(sys.stdin.readline())
card_cnt = list(map(int, sys.stdin.readline().split()))

card = Counter(card) # Counter 함수를 통해 각각의 숫자가 몇 개씩 있는지 세준다
print(card)
for i in card_cnt:
    print(card[i], end = " ")