[백준] 10816번 숫자 카드2 - Java

[백준] 10816번 숫자 카드2 - Java

 

문제 출처

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

10816번: 숫자 카드 2

 

---

※ 풀이

N과 M의 범위가 매우 넓으므로 그냥 탐색하면 시간 초과가 난다.

따라서 일반적인 풀이는 이분탐색을 통해 데이터를 탐색하는 문제이다.

 

하지만 이분탐색을 쓰지 않고 HashMap 으로 풀어보았다.

숫자를 입력받으면서 <key: 수, value: 해당 수가 등장한 횟수> 형식으로 카운팅되어 들어가고

입력이 끝나면 출력이 되는 형태이다.

 

---

※ 소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;

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());

        String[] split = br.readLine().split(" ");
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            int number = Integer.parseInt(split[i]);
            if(map.containsKey(number)){
                map.replace(number, map.get(number) + 1);
            }
            else{
                map.put(number, 1);
            }
        }

        StringBuilder stb = new StringBuilder();
        int m = Integer.parseInt(br.readLine());
        split = br.readLine().split(" ");
        for (int i = 0; i < m; i++) {
            int number = Integer.parseInt(split[i]);
            if (map.containsKey(number)) {
                stb.append(map.get(number));
            }
            else{
                stb.append(0);
            }

            if (i < m - 1) {
                stb.append(" ");
            }
        }
        System.out.println(stb);
    }
}