[Algorithm] 🎴백준 10816 숫자 카드 2

0. 문제

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 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. 아이디어

HashMap 이용
💡 카드의 숫자를 key값으로 이용
💡 처음 나온 숫자는 value에 1 저장, 이미 나왔던 숫자는 value에 value+1을 저장
💡 m의 숫자가 key 값으로 있는지 확인 후에 value 출력

2. 핵심 풀이

1) 처음 나온 숫자는 value에 1 저장, 이미 나왔던 숫자는 value에 value+1을 저장

if(map.containsKey(num)) {
	map.put(num,map.get(num)+1);
} else {
	map.put(num, 1);
}

2) m의 숫자가 key 값으로 있는지 확인 후에 value 출력

if(map.containsKey(num))
	sb.append(map.get(num));
else
	sb.append("0");
sb.append(" ");

3. 코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashMap;
public class Hash_3 {

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int n = Integer.parseInt(br.readLine());
		HashMap<Integer, Integer> map = new HashMap<>();

		String[] s = br.readLine().split(" ");
		
		for(int i=0; i<n; i++) {
			int num = Integer.parseInt(s[i]);
			
			if(map.containsKey(num)) {
				map.put(num,map.get(num)+1);
			} else {
				map.put(num, 1);
			}
		}
		
		int m = Integer.parseInt(br.readLine());
		s = br.readLine().split(" ");
		StringBuilder sb = new StringBuilder();
		
		for(int i=0; i<m; i++) {
			int num = Integer.parseInt(s[i]);
			
			if(map.containsKey(num))
				sb.append(map.get(num));
			else
				sb.append("0");
			sb.append(" ");
		}
		
		System.out.println(sb.toString());
	}

}

4. 결과

성공✨

처음에는 System.out.println(map.get(num)+" ") 했다가 시간초과가 났다.
+" " 연산은 시간을 많이 잡아먹는다고 한다.
이제부터 StringBuilder나 BufferedWriter를 잘 활용해야 겠다...