[BaekJoon] 1620번 나는야 포켓몬 마스터 이다솜 문제 풀이 (Success)

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

 

☕Language: Java

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.valueOf(st.nextToken());
        int M = Integer.valueOf(st.nextToken());
 
        Map<String, Integer> map = new HashMap<>();
        List<String> list = new ArrayList<>();
 
        for (int i = 1; i <= N; i++) {
            String s = br.readLine();
            map.put(s, i);
            list.add(s);
        }
 
        StringBuilder sb = new StringBuilder();
 
        for (int i = 0; i < M; i++) {
            String s = null;
            try {
                s = br.readLine();
                int num = Integer.valueOf(s);
                sb.append(list.get(num - 1)).append("\n");
            } catch (NumberFormatException nfe) {
                sb.append(map.get(s)).append("\n");
            }
        }
 
        bw.write(sb.toString());
 
        bw.flush();
        bw.close();
 
    }
}
 

🤔 해설

1. sb.append(list.get(num - 1)).append("\n");

    - 입력값을 숫자로 바꿔서 nfe가 발생하지 않으면 list에서 String 출력

2. catch (NumberFormatException nfe) { ... }

    - 입력값을 숫자로 바꿔서 nfe가 발생하면 map에서 value 출력

 

😮 이 외의 풀이

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashMap;
import java.util.Map;
 
public class Main {
 
    static StringBuilder sb;
 
    static Map<Integer, String> mapIntStr;
    static Map<String, Integer> mapStrInt;
 
    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
        String[] NM = br.readLine().split(" ");
 
        int N = Integer.valueOf(NM[0]);
        int M = Integer.valueOf(NM[1]);
 
        mapIntStr = new HashMap<>();
        mapStrInt = new HashMap<>();
 
        for (int i = 1; i <= N; i++) {
            String name = br.readLine();
            mapIntStr.put(i, name);
            mapStrInt.put(name, i);
        }
 
        sb = new StringBuilder();
 
        for (int i = 1; i <= M; i++) {
            String str = br.readLine();
            result(str);
        }
 
        bw.write(sb.toString());
        bw.flush();
        bw.close();
    }
 
    public static void result(String str) {
        if (Character.isDigit(str.charAt(0))) {
            sb.append(mapIntStr.get(Integer.parseInt(str)) + "\n");
        } else {
            sb.append(mapStrInt.get(str) + "\n");
        }
    }
 
}
 

1. String[] NM = br.readLine().split(" ");

    - BufferedReader의 split 사용

2. mapIntStr, mapStrInt

    - Key-Value 타입을 달리해서 map에 각각 저장

3. if (Character.isDigit(str.charAt(0))) { ... }

    - Character 함수 중 숫자 판별 함수 사용

    - 1번째 글자만으로도 숫자인지 판별할 수 있으므로 charAt에 0 대입

 

 

 

🔗 소스 코드
GitHub

 

📚 참고 자료

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