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

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

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

 

• 대화가 앞쪽에 있는데 문제랑 별로 상관없는 대사들이다...
• name과 index 처리를 위한 HashMap 두개를 만들어서 키 검사 후 있으면 출력했다.

 

import java.io.*;
import java.util.*;


public class Main {
    public static final BufferedReader br = new BufferedReader(new InputStreamReader(System.in));


    public static void main(String[] args) throws Exception {
        StringTokenizer st = new StringTokenizer(br.readLine());

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

        HashMap<String, String> nameMap = new HashMap<>();
        HashMap<String, String> integerMap = new HashMap<>();

        for (int i = 1; i <= N; i++) {
            String name = br.readLine();
            nameMap.put(name, String.valueOf(i));
            integerMap.put(String.valueOf(i), name);
        }

        StringBuilder builder = new StringBuilder();

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

            if(integerMap.containsKey(keyWord)) {
                builder.append(integerMap.get(keyWord) + "\n");
            }

            if(nameMap.containsKey(keyWord)) {
                builder.append(nameMap.get(keyWord) + "\n");
            }
        }

        System.out.println(builder);
    }
}