백준 10816번 - 숫자 카드 2 - 파이썬(Python)

문제 링크

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

 

문제

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 몇 개 가지고 있는지 구하는 프로그램을 작성하시오.

입력 조건

• 첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.
• 셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 몇 개 가지고 있는 숫자 카드인지 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.

출력 조건

• 첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 몇 개 가지고 있는지를 공백으로 구분해 출력한다.

 

입력 예시1

10
6 3 2 10 10 10 -10 -10 7 3
8
10 9 -5 2 3 4 5 -10

출력 예시1

3 0 0 1 2 0 0 2

 

풀이 전 생각

https://js-kkk.tistory.com/14

백준 3078번 - 좋은 친구 - 파이썬(Python)

이 문제가 떠올랐다. 

 

메모리를 많이 차지하겠지만, 숫자 카드에 적혀있는 수의 범위를 '-10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.'로 제한했기 때문에

 

1. 숫자의 값을 인덱스로 가지게 해서 count해줄 수 있도록 크기가 20_000_001 인 배열을 선언해서

2.입력받은 수를 count해주고, 그리고 count를 출력해야할 숫자들을 출력해주면 될 것 같다.

 

하지만 메모리를 너무 많이 차지하기 때문에 다른 방식으로도 접근해봐야겠다.

 

 

예제 코드

n = int(input())
num1 = list(map(int,input().split()))
m = int(input())
num2 = list(map(int,input().split()))

cnt = [0]*20_000_001

for i in num1:
    cnt[i]+=1

for j in num2:
    print(cnt[j], end=' ')

 

 

생각