[JAVA]백준 알고리즘 10816번 : 숫자 카드 2

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

10816번: 숫자 카드 2

문제

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

 

풀이

HashMap을 이용하여 해결했다.

카드의 숫자가 HashMap의 key값에 존재하면, 해당 key값의 value를 1 증가시키고,

카드의 숫자가 HashMap의 key값에 존재하지 않으면, 해당 key값의 value를 1로 저장하였다.

 

코드

import java.io.*;
import java.util.HashMap;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuilder sb = new StringBuilder();

        int N = Integer.parseInt(br.readLine());
        HashMap<Integer, Integer> card = new HashMap<>();

        st = new StringTokenizer(br.readLine());
        for(int i = 0; i < N; i++) {
            int inputNum = Integer.parseInt(st.nextToken());
            if(card.containsKey(inputNum)) {
                int count = card.get(inputNum);
                card.put(inputNum, count + 1);
            } else {
                card.put(inputNum, 1);
            }
        }

        int M = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());
        for(int i = 0; i < M; i++) {
            int num = Integer.parseInt(st.nextToken());
            if(card.containsKey(num)) sb.append(card.get(num)).append(" ");
            else sb.append(0).append(" ");
        }
        System.out.println(sb);
    }
}

틀린 부분이 있다면 정정해주시면 감사하겠습니다.
궁금한 부분이 있거나, 다른 아이디어가 있으시면 자유롭게 댓글 남겨주세요!