[백준] 10816번 - 숫자 카드 2 (Silver 4)

문제 링크

백준 10816번 - 숫자 카드 2 (Silver 4)

문제 설명

입력받은 M개의 카드가 먼저 입력받은 N개의 카드 중에 몇 개 있는지를 찾는 문제

맵(Map)

(key, value) 쌍으로 이루어진 자료구조이다.
이 문제에서는 key를 카드 번호로, value를 key 카드가 몇 장 들어있는지로 활용할 수 있다.
풀이에 사용된 HashMap의 경우, key의 탐색에 $O(1)$, 새로운 데이터 추가에도 $O(1)$이 소요된다.

정답 코드 및 설명

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

public class Main {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int N = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        HashMap<Integer, Integer> cards = new HashMap<>();
        for (int i = 0; i < N; i++) {
            int newCard = Integer.parseInt(st.nextToken());
            if (cards.containsKey(newCard))
                cards.replace(newCard, cards.get(newCard) + 1);
            else
                cards.put(newCard, 1);
        }

        int M = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < M; i++) {
            int cardNum = Integer.parseInt(st.nextToken());
            if (cards.get(cardNum) == null)
                bw.write("0 ");
            else
                bw.write(cards.get(cardNum) + " ");
        }
        bw.close();
    }
}

• N장의 카드 입력 받기
  • 이미 있는 카드 번호가 들어오면 value를 1 증가 시킴
  • 없는 카드 번호이면 value = 1로 추가
• 카드 장수 출력
  • 없는 카드 번호이면 get(cardNum)이 null을 반환한다는 사실을 활용함