[BOJ] 나는야 포켓몬 마스터 이다솜

[1620] 나는야 포켓몬 마스터 이다솜

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

• 포켓몬 도감에서 포켓몬의 이름을 보면 포켓몬의 번호를 말하거나, 포켓몬의 번호를 말하면 포켓몬 이름을 말하는 는 프로그램

 

 

Solution

• HashSet 두 개를 만들어 문자가 입력되면 숫자 출력, 숫자가 입력되면 문자가 출력되도록 구현

 

 

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.StringTokenizer;
 
public class Main {
    static int n, m;
    static HashMap<String, Integer> word;
    static HashMap<Integer, String> number;
 
    static StringTokenizer st;
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        st = new StringTokenizer(br.readLine().trim());
        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());
        init();
        String pocketmon;
        for (int i = 1; i <= n; i++) {
            pocketmon = br.readLine().trim();
            word.put(pocketmon, i);
            number.put(i, pocketmon);
        }
 
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < m; i++) {
            String quiz = br.readLine();
            if (Character.isDigit(quiz.charAt(0))) {
                sb.append(number.get(Integer.parseInt(quiz)) + '\n');
            } else {
                sb.append(Integer.toString(word.get(quiz)) + '\n');
            }
        }
        System.out.println(sb.toString());
 
    }
 
    public static void init() {
        word = new HashMap<>();
        number = new HashMap<>();
    }
}