[Java] BOJ 1620 나는야 포켓몬 마스터 이다솜

문제 

문제 링크 https://www.acmicpc.net/problem/1620

• 첫째 줄에는 도감에 수록되어 있는 포켓몬의 개수 N이랑 맞춰야 하는 문제의 개수 M이 주어져. N과 M은 1보다 크거나 같고, 100,000보다 작거나 같은 자연수
• 포켓몬의 이름은 모두 영어로만 이루어져있음 (첫 글자만 대문자이고, 나머지 문자는 소문자로만 이루어져 있으나 일부 포켓몬은 마지막 문자만 대문자일 수도 있음)
• 포켓몬 이름의 최대 길이는 20, 최소 길이는 2
• 그 다음 줄부터 총 M개의 줄에 맞춰야하는 문제가 입력으로 들어옴
• 문제가 알파벳으로만 들어오면 포켓몬 번호를 말해야 하고, 숫자로만 들어오면, 포켓몬 번호에 해당하는 문자를 출력해야함
• 입력으로 들어오는 숫자는 반드시 1보다 크거나 같거나 N보다 작거나 같음
• 입력으로 들어오는 문자는 반드시 도감에 있는 포켓몬의 이름만 주어짐 

아이디어

• hashmap을 사용하여 key값으로 value를 찾는 방법을 채택했다.
• 포켓몬 이름인 문자열로 번호를 찾을 수도 있고 포켓몬 번호인 숫자로 찾을 수도 있기 때문에 hashmap을 두번 써줬다.(key 값이 String 인 hashmap과 Integer인 hashmap)
• 입력값을 무조건 String으로 받기에 (내 방법은 그렇다..) key 값이 문자열인지 숫자인지 구분하기 위해 문자열 제일 앞부분을 문자로 바꿔서 아스키코드로 대문자인지 감별했다.

겪은 시행착오

• 처음에는 hashmap을 생각하지 못하고 반복문을 사용해서 문자열을 비교해서 찾는 방법을 사용해, 시간초과가 났다..... 

혹시 java 8로 하면 시간이 줄어들까해서 돌려보았다.. (부끄럽다)

 

시간 초과가 난다면 꼼수 부릴 생각하지 말고 다시 제대로 풀어보자....

코드

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

// hashmap 사용
// 데이터를 저장할때 key와 value가 짝을 이루어 저장됨
// 키 값으로 hash 함수를 실행한 결과를 통해 저장위치를 결정
// 추가, 삭제, 특히 검색이 빠름

public class BOJ1620 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());

        // 포켓몬 도감 수
        int list = Integer.parseInt(st.nextToken());
        // 맞춰야 되는 문제 수
        int test = Integer.parseInt(st.nextToken());

        // Hashmap에 key값과 value 값을 사용하여 저장
        // 이름으로 찾는 경우와 번호로 찾는 경우 두가지가 있기 때문에 두번 저장
        HashMap<String, Integer> poket = new HashMap<String, Integer>(list);
        HashMap<Integer, String> poket2 = new HashMap<Integer, String>(list);

        for(int i = 1; i < list + 1; i++){
            // 도감에 등록될 포켓몬 이름
            String s = br.readLine();

            // 도감에 등록
            poket.put(s, i);
            poket2.put(i, s);
        }

        for(int i = 0; i < test; i++){
            String s = br.readLine();

            // 테스트하는 문제가 번호인지 이름인지 알기 위해서
            // 첫 문자를 떼서 아스키코드로 비교하기 위해 변환
            int ch = (int)s.charAt(0);

            // 아스키코드로 비교해서 대문자일 경우,
            // 도감에서 key 값인 이름으로 value 값인 번호 찾아서 저장
            if((65 <= ch) && (ch <= 90)){
                int n = poket.get(s);
                sb.append(n + "\n");
            }
            // 첫 문자가 대문자가 아닐 경우,
            else{
                // 문자열로 받았으므로 int로 변환
                int num = Integer.parseInt(s);
                // 도감에서 key 값인 번호로 value 값인 해당 포켓몬 이름을 찾아 저장
                String po = poket2.get(num);
                sb.append(po + "\n");
            }
        }

        System.out.println(sb);
    }
}