백준 1620 나는야 포켓몬 마스터 이다솜 | C++

 

 

이번 포스팅은 백준 1620번 나는야 포켓몬 마스터 이다솜입니다.

아래 url를 클릭하시면 백준 사이트에서 문제를 볼 수 있습니다.

www.acmicpc.net/problem/1620

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

 

 

 

기본 알고리즘

unordered_map (hash_map)

 

참고

백준에서 hash_map 사용시 컴파일 에러가 발생한다.

hash_map → unordered_map 으로 변경하면 해결된다.

 

 

전체 코드

 

//백준1620 나는야 포켓몬 마스터 이다솜                                        
 
#include <iostream>
#include <unordered_map>
#include <string>
using namespace std;
 
int main() {
    int N, M;
    unordered_map<int, string> m1;
    unordered_map<string, int> m2;
 
    cin >> N >> M;
    for (int i = 1; i <= N; i++) {
        string name;
        cin >> name;
        m1.insert(pair<int, string>(i, name));
        m2.insert(pair<string, int>(name, i));
    }
 
    string* quiz = new string[M];
    string* ans = new string[M];
    for (int i = 0; i < M; i++) {
        cin >> quiz[i];
    }
 
    for (int i = 0; i < M; i++) {
        if (quiz[i].at(0) < 'A') { //숫자
            int n = stoi(quiz[i]);
            ans[i] = m1.find(n)->second;
        }
        else { //이름
            int m = m2.find(quiz[i])->second;
            ans[i] = to_string(m);
        }
    }
 
    for (int i = 0; i < M; i++) {
        if (ans[i].at(0) < 'A') { //숫자
            printf("%d\n", stoi(ans[i]));
        }
        else { //이름
            cout<<ans[i]<<"\n";
        }
    }
}