[BOJ] [JAVA] 10816번 숫자 카드 2

두번째 숫자 카드 문제다.

10816번: 숫자 카드 2

이 문제에서는 HashMap의 또다른 사용 법을 알 수 있었다..!

HashMap은 key값을 검색했을 때 존재하지 않는 key라면 null을 반환한다.

이 점을 이용해서 cardMap에 카드의 개수가  바로바로 value값으로 입력되도록 했다.

이 후 key를 검색해서 value(동일한 수를 가진 카드의 개수)를 출력했다.

import java.util.*;
import java.io.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		
		HashMap <Integer, Integer> cardMap = new HashMap<>();
		StringTokenizer st = new StringTokenizer(br.readLine());
		for (int i = 0; i < n; i++) {
			int a = Integer.parseInt(st.nextToken());
			if (cardMap.get(a) == null) {
				cardMap.put(a,  1);
			} else cardMap.put(a, cardMap.get(a) + 1);
		}
		
		int m = Integer.parseInt(br.readLine());
		StringTokenizer st1 = new StringTokenizer(br.readLine());
		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < m; i++) {
			int b = Integer.parseInt(st1.nextToken());
			if (cardMap.get(b) == null) sb.append(0).append(" ");
			else sb.append(cardMap.get(b)).append(" ");
		}
		System.out.print(sb);
	}
}

백준 단계별로 풀어보기 : 집합과 맵 부분을 풀고 있는데

새로운 개념들을 많이 배웠다..!