[BOJ] 백준 [10816] 숫자카드2 JAVA

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

10816번: 숫자 카드 2

문제

풀이

TreeMap 으로 이진탐색을하여 이미 존재하는 숫자면 replace로 기존값 +1

처음 들어오는 숫자면 put을 하여 구성을 한 후 , containkey 함수를 호출하여 각 값을 출력하면된다.

print함수를 사용하면 최악의 경우 50만번 출력을 해서 시간초과가 났었다.

StringBuilder를 사용하여 한번에 출력할 것.

 

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));br.readLine();
        TreeMap<Integer, Integer> map = new TreeMap<>();
        Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt)
                .forEach(e->{
                    if(map.containsKey(e))
                        map.replace(e,map.get(e)+1);
                    else
                        map.put(e,1);
                });br.readLine();
        StringBuilder sb = new StringBuilder();
        Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt)
                .forEach(e->{
                    sb.append(map.getOrDefault(e, 0)).append(" ");
                });
        System.out.println(sb.toString());
    }
}