백준 1620 나는야 포켓몬 마스터 이다솜(JAVA)

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

1620번: 나는야 포켓몬 마스터 이다솜

 

HashMap 하나로만 해결하려하니 시간초과가 계속 나오는 문제였습니다.

구글링을 통해 도감에 넣고, 도감에서 정보를 불러올 때 숫자인지, 문자열인지 판단하여 keySet() 이나 entrySet같은 HashMap의 메소드를 사용하지 않고 구현해야 했습니다.

 

 

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));
        HashMap<String, Integer> map = new HashMap<>();
 
        StringTokenizer st = new StringTokenizer(br.readLine()," ");
        
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
 
        StringBuilder sb = new StringBuilder();
        String nameArr[] = new String[N+1];
        
        for(int i=1; i<N+1; i++) {
            String s = br.readLine();
            map.put(s, i);
            nameArr[i] = s;
        }
        
        while(M-- > 0) {
            String str = br.readLine();
            if(CheckName(str)) {
                int idx = Integer.parseInt(str);
                sb.append(nameArr[idx]);
            } else {
                sb.append(map.get(str));
            }
            sb.append("\n");
        }
        
        System.out.println(sb);
        br.close();
    }
 
    private static boolean CheckName(String s) {
        try {
            Double.parseDouble(s);
            return true;
        } catch (NumberFormatException e) {
            return false;
        }
    }
}