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

문제 설명

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

 

풀이과정

 Map 자료구조를 이용해서 쉽게 풀이할 수 있다.

 

정답코드

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));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        Map<String, Integer> map = new HashMap<>();
        Map<Integer, String> map2 = new HashMap<>();


        for (int i = 0; i < N; i++) {
            String name = br.readLine();
            map.put(name, i + 1);
            map2.put(i + 1, name);
        }

        for (int i = 0; i < M; i++) {
            String cmd = br.readLine();

            if (Character.isDigit(cmd.charAt(0))) {
                bw.write(map2.get(Integer.parseInt(cmd)) + "\n");
            } else {
                bw.write(map.get(cmd) + "\n");
            }
        }
        
        bw.flush();
    }
}