백준 1620 자바

<문제>

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

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

 

<풀이>

두개의 HashMap을 이용하면 편하다. 

하나는 key값이 포켓몬 이름이고 value가 숫자인 HashMap을 만들고 다른 하나는 key값이 숫자이고 Value값이 포켓몬 이름인 HashMap을 만든다.

입력한 글자의 첫글자를 기준으로 숫자인지 아닌지 확인한다. 만약 숫자일경우 숫자에 대응하는 포켓몬이름을 가져오면 되고 숫자가 아닐경우 포켓몬이름에 맞는 숫자를 가져오면 된다.

출력할때 StringBuilder 에 값을 넣어서 한번에 출력을 했다.

 

<코드>

import java.util.*;
import java.io.*;
public class Main {

    public static void main(String[] args)throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        StringBuilder sb = new StringBuilder();
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        HashMap<Integer,String> hashMap1 = new HashMap<>();
        HashMap<String, Integer> hashMap2 = new HashMap<>();

        for(int i = 0; i < n; i++) {
            String s = br.readLine();
            hashMap1.put(i+1,s);
            hashMap2.put(s,i+1);
        }

        for(int i = 0; i < m; i++) {
            String s = br.readLine();
            if(Character.isDigit(s.charAt(0))) {
                sb.append(hashMap1.get(Integer.parseInt(s))+"\n");
            }
            else{
                sb.append((hashMap2.get(s))+"\n");
            }
        }
        System.out.println(sb);
    }
}