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

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

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

풀이

이 문제는 자바의 해시맵으로 간단히 구현할수있다. 우선 입력값을 차례대로 해시맵에 넣고 그다음 문제에서 주는 key값과 value값에 알맞게 출력하면된다. 여기서 문제는 key값을 주어졌을 경우는 hashMap의 get을 이용해 꺼내면 되지만 value가 주어졌을경우는 따로 할수있는방법이 없다 물론 key set을 꺼내어 하나씩 맞추어 가면서 출력하면 시간초과로 문제를 틀리게 된다. 그렇기에 우리는 처음에 해시맵에 넣을때 따로 String배열을 만들어 저장하면된다. 그렇게 하면 순차적으로 검색하면서 하지않아도 된다. 즉 hashMap의 key값을 String으로 하고 value를 Integet로 하고 String 의 배열을 선언하면 된다. 그리 어렵지 않은 문제 이기에 코드를 보면 이해가 빠를듯하다.

 

코드

import java.io.*;
import java.util.HashMap;
import java.util.StringTokenizer;

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());
        HashMap<String , Integer> hm = new HashMap<String, Integer>(); //문자로 들어 올 경우
        String[] arr;


        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        arr = new String[N+1]; // 숫자로 들어올 경우


        for (int i = 1; i <= N; i++){
            String temp = br.readLine();
            hm.put(temp, i);
            arr[i] = temp;
        }

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

            if (hm.containsKey(temp)) //문자로 들어올 경우
                bw.write(hm.get(temp) + "\n");

            else { //숫자로 들어올경우
                bw.write(arr[Integer.parseInt(temp)] + "\n");
            }

        }
        bw.flush();
        bw.close();
        br.close();
    }
}