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

문제 출처 :

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

알고리즘 분석 :

문제 해결에 필요한 사항

1. Map STL :: http://www.crocus.co.kr/604

Map을 이용하면 해결할 수 있다.

그냥 탐색을 통해 찾는다면 O(n^2)으로 TLE를 받게 된다.

물론 소스 코드도 길어질 것이다.

소스 코드 : 

#include <iostream>
#include <map>
#include <string>
#include <cstdio>
#include <string.h>
 
using namespace std;
 
// 문자열로 제작된 map
map<string, int> m1;
// 숫자로 제작된 map
map<int, string> m2;
 
int main()
{
    int n, k;
    char tmp[22];
    int get;
    string str;
 
    scanf("%d %d", &n, &k);
 
    // 포켓몬 이름에 대한 포켓몬 번호를 m1에 저장해주고
    // 포켓몬 번호에 대한 포켓몬 이름을 m2에 저장한다.
    for (int i = 1; i <= n; i++)
    {
        scanf("%s", tmp);
        m1[tmp] = i;
        m2[i] = tmp;
    }
 
    for (int i = 0; i < k; i++)
    {
        scanf("%s", tmp);
        str = tmp;
        // 숫자라면 문자를 숫자로 바꾸어 탐색 후 출력
        if (48 <= tmp[0] && tmp[0] <= 57)
            printf("%s\n", m2[atoi(tmp)].c_str());
        // 문자라면 해당하는 값을 출력
        else
            printf("%d\n", m1[str.c_str()]);
    }
}
 
//                                                       This source code Copyright belongs to Crocus
//                                                        If you want to see more? click here >>