[백준] 1620번 - 나는야 포켓몬 마스터 이다솜 [Java][C++]

문제 링크

---

1. 문제 풀이

번호가 $1$ 인 포켓몬부터 이름이 주어진 후, 번호가 입력으로 들어오면 포켓몬의 이름을, 이름이 입력으로 들어오면 포켓몬의 번호를 출력하는 문제로 해시맵을 활용하면 간단하게 해결할 수 있다.

포켓몬의 이름은 모두 영어로 되어 있으므로 해시맵에 포켓몬의 이름을 키로, 포켓몬의 번호를 값으로 한 쌍 저장하고, 포켓몬의 번호를 키로, 포켓몬의 이름을 값으로 한 쌍 저장하면 해시맵 하나로 쿼리들을 처리할 수 있다. 이때 하나의 해시맵에서 처리하기 위해 포켓몬의 번호도 문자열 타입으로 맞춰줬다.

---

2. 코드

1. 해시를 사용한 집합과 맵 [Java]

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

public 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());

        Map<String, String> map = new HashMap<>();

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        for (int i = 1; i <= N; i++) {
            String name = br.readLine();
            map.put(String.valueOf(i), name);
            map.put(name, String.valueOf(i));
        }

        for (int i = 0; i < M; i++) {
            String command = br.readLine();
            bw.write(map.get(command));
            bw.newLine();
        }

        bw.flush();
    }
}

2. 해시를 사용한 집합과 맵 [C++]

#include <bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    unordered_map<string, string> mp;

    int n, m;
    cin >> n >> m;

    for (int i = 1; i <= n; i++) {
        string name;
        cin >> name;

        mp[to_string(i)] = name;
        mp[name] = to_string(i);
    }

    for (int i = 0; i < m; i++) {
        string s;
        cin >> s;
        cout << mp[s] << '\n';
    }
}

---

3. 풀이 정보

1. 해시를 사용한 집합과 맵 [Java]

언어	시간	메모리	코드 길이
Java 11	464 ms	49944 KB	878 B

2. 해시를 사용한 집합과 맵 [C++]

언어	시간	메모리	코드 길이
C++ 17	132 ms	31440 KB	449 B

---