백준: 10816 숫자 카드 2

문제

10816번: 숫자 카드 2

 

문제 풀이

앞서 다룬 숫자 카드 문제와 달리, M개의 카드 중에 N개의 카드에 포함 되는 경우 카드의 개수를 출력하는 문제이다. 따라서 N개의 카드의 중복을 `set`으로 중복 여부를 없애는 것이 아닌, `Counter`를 통해 각 카드가 몇 개씩 존재하는지 확인하면 된다.

 

코드

from sys import stdin
from collections import Counter


if __name__ == "__main__":
    n = int(stdin.readline())
    n_card = Counter(list(map(int, stdin.readline().split())))
    m = int(stdin.readline())
    m_card = list(map(int, stdin.readline().split()))

    for card in m_card:
        if card in n_card:
            print(n_card[card], end=' ')
        else:
            print(0, end=' ')