[JS] 백준 10816번 : 숫자 카드 2

 

 

Question

백준 10816번 : 숫자 카드2

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

 

10816번: 숫자 카드 2

 

 

입력

첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.

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

 

출력

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

 

입력예시

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

 

출력예시

3 0 0 1 2 0 0 2

 

 

---

 

 

My Code

 

 

처음에는 set으로 풀려고 했으나 set은 중복을 허용하지 않아 카드의 개수를 셀 수가 없었다.

고로 새로운 자료형이 필요했다.

배열을 이용해서 풀려고 했지만 딱 봐도 시간 초과를 하고 틀릴게 눈에 보였다. <한 번 속지 두 번 속냐 사실 속았다 🤷‍♂️

 

 

그리고 node.js 입력받는 부분에서 자꾸 시간초과가 떠서 여기저기 구글링 좀 했다.

입력받는 것만 좀... 해결되면 스트레스가 반으로 줄 거 같다.

잘 풀었는데 안 나와서 모니터 주먹으로 치고 싶었음.

 

 

const input = require("fs").readFileSync("dev/stdin").toString().trim().split(/\s/);
const n = +input[0];

const nArr = input.slice(1, n + 1).map(Number);
const mArr = input.slice(n + 2).map(Number);

const map = new Map();
nArr.forEach((ele) => {
    if (map.has(ele)) { map.set(ele, map.get(ele) + 1) } else { map.set(ele, 1) }
})
let result = [];

mArr.forEach((ele) => {
    if (map.has(ele)) { result.push(map.get(ele)) } else { result.push(0) }
})


console.log(result.join(' '));

맞았습니다!!

 

 

 

HOW? (풀이 방법)

 

• map을 이용해서 카드 종류는 key, 카드의 장수를 value값으로 저장한다.
• map이 key를 가지고 있으면 value 값을 +1 해준다
• 원래 갖고 있는 카드인 nArr을 map으로 만들어준다.
• 비교하려는 카드인 mArr의 각 element가 map에 있는지 확인하고 있으면 몇 장 갖고 있는지를 result배열에 push 한다.

 

 

 

배열을 순회하는 것보다 map을 순회하는 게 더 시간을 단축하는구나

hash를 이용해서 풀라고 하던데 아직 이해가 잘 안 간다.

자료구조라면 찾아보고 정리 해야겠다.