스파르탄 365 2주차 (4) 숫자 카드 2

2주차

백준 10816번 제로

문제링크 : https://www.acmicpc.net/problem/10816

💡 풀이 전 계획과 생각

1. 앞 문제에서 for문을 써 시간 초과가 났으니 스택을 사용하여 비교를 해보자
   -> 시간초과!

💡 풀이

import sys
from collections import Counter

N = int(sys.stdin.readline())
cards = list(map(int, sys.stdin.readline().split()))
M = int(sys.stdin.readline())
finded_list = list(map(int, sys.stdin.readline().split()))

counter = Counter(cards)

print(' '.join(str(counter[c]) if c in counter else '0' for c in finded_list))

🧐 막혔던 점과 고민

1. 스택을 써도 시간초과

스택 말고는 더 빠르게 비교를 할 방법이 생각이 나지 않아 너무 당황했었다.

👏🏻 알게된 개념과 소감

• from collections import Counter
  Counter 클래스를 통해 더 빠른 비교를 할 수 있음을 알았다.

  https://velog.io/@dawnofspring/파이썬-collections-모듈의-Counter-클래스

• 덕분에 join 함수를 더 잘 이해하여 사용할 수 있게 되었다.

counter = Counter(cards)
print(' '.join(str(counter[c]) if c in counter else '0' for c in finded_list))