백준 1620 <나는야 포켓몬 마스터 이다솜>

1620번: 나는야 포켓몬 마스터 이다솜 (acmicpc.net)

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

다솜이가 포켓몬을 참 좋아하는갑다.

문제를 상당히 정성스럽게 만드셨다. 일단  문제를 보자마자 map을 만들어야겠다는 생각을했고

문제의 조건은 input값이 String일때 (포켓몬이름) 포켓몬 번호를 출력하고

input값이 Integer일때 (포켓몬 번호)일때 포켓몬 이름을 출력하라고 했다.

 

일반적으로 map을 사용할때, key > value를 얻는건 매우쉽다. 그냥 

if(containsKey(key1) > get(key1) 해버리면 value를 쉽게 얻지만,

그 반대로 value에서 key를 끌어오는건 좀 더 복잡하다.

 

이 문제는 만약 내가 map(Integer, String)으로 설정하고 문제를  시작한다면 위에 말했던 반대로 끌어는 코드를 구현하는건 불가피하다. 
따라서 맵을 두개 사용하는 생각을 해보면 문제가 매우 간단해진다.

 

Map<Integer,String> 한개, Map<String,Integer> 한개 설정해버리면 문제의 조건따라 포켓몬 이름이 들어오면 두번째 map을 써서 value바로 얻고, 그 반대면 첫번째 맵을 쓰면 끝.

 

이 아이디어만 알면 바로 코드로 구현할수 있겠다.

 

bufferdreader을 썼기에 try -catch 구문을 써서 numberformat 예외 처리를 해줬다. 문제에서 포켓몬 이름이 들어올 경우 parseInt()가 안되니까 !

package 백준;
import java.io.*;

import java.util.*;
public class _1620_포켓몬마스터_시간절약코드 {

	public static void main(String[] args) throws IOException {
		
		//이렇게하면 메소드를 안쓰고 가능 즉 map을 2가지 종류로 만들어버리기.
		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<Integer, String> numToName = new HashMap<>();
        HashMap<String, Integer> nameToNum = new HashMap<>();

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

        for (int i = 0; i < m; i++) {
            String question = br.readLine();

            try {
                int numQuestion = Integer.parseInt(question);
                System.out.println(numToName.get(numQuestion));
            } catch (NumberFormatException e) {
                String nameQuestion = question;
                System.out.println(nameToNum.get(nameQuestion));
            }
        }
    

	}

}