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

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

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

 

 

풀이)

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.StringTokenizer;

public class Main_BJ_1620_나는야포켓몬마스터이다솜 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        HashMap<String, Integer> nameHash = new HashMap<>();
        HashMap<Integer, String> indexHash = new HashMap<>();

        for(int i=1; i<=n; i++) {
            String name = br.readLine();
            nameHash.put(name, i);
            indexHash.put(i, name);
        }

        for(int i=0; i<m; i++){
            String check = br.readLine();
            //숫자이면
            if(Character.isDigit(check.charAt(0))){
                System.out.println(indexHash.get(Integer.parseInt(check)));
            }
            //문자열이면
            else{
                System.out.println(nameHash.get(check));
            }
        }


    }//main
}

일단 문제가 너무 웃겼다.ㅋㅋㅋㅋㅋㅋㅋ

그냥 윗 부분은 안 읽고 밑의 조건만 확인해도 충분이 이해가 가능한 문제!

빠르게 검색하기 위해 hash를 두 개 사용해서 만들었다.

또한, 문자열인지 숫자인지 확인하기 위해 Character.isDigit()을 활용했다!

 

이것만 안다면 간단한 문제~~