[BOJ] 백준 10816 숫자 카드 2 / 이분탐색

이미지를 클릭하면 문제 사이트로 이동합니다.

 

 

 

💭 문제 이해

 

이전에 풀었던 숫자 카드 문제와 거의 비슷하다.

주어진 숫자 입력들이 상근이가 가진 숫자 카드에 있는지 여부를 체크하는 문제였다면,

 

이 문제는 상근이가 그 숫자들을 몇 개 가지고 있는 지도 체크해야 한다는 점이다.

 

Python의 Counter를 이용해서 상근이의 숫자 카드를 Key 값으로 카드번호, Value 를 카드 개수로 하는 딕셔너리를 생성했다.

그리고 입력으로 들어온 숫자가 Counter 객체에 존재한다면 Value를 출력하고, 없다면 0을 출력하게 구현했다.

 

 

 

구현 언어: Python

 

import sys
from collections import Counter

r = sys.stdin.readline

answer = []
n, my_card = int(r()), Counter(map(int, r().split()))
m, num_list = int(r()), list(map(int, r().split()))

for num in num_list:
    answer.append(my_card[num] if num in my_card.keys() else 0)
print(*answer)

 

 

여기서 더 Pythonic 하고 싶다면,

import sys
from collections import Counter
r = sys.stdin.readline

_, my_card, _ = r(), Counter(r().split()), r()
print(" ".join(str(my_card[num]) if num in my_card else '0' for num in r().split()))

 

요런 것도 가능하다 🎉