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

www.acmicpc.net/problem/1620

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

풀었다고 하기도 뭐한데 맵 하나만 사용하면 tle가 떠서 맵을 두개 썼다..

#include <unordered_map>
#include <iostream>
#include <string>
using namespace std;

int n, m;
string tmp;

unordered_map<string, int> um;
unordered_map<int, string> um2;
int main(void)
{
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  cin >> n >> m;
  for (int i = 1; i <= n; i++)
  {
    cin >> tmp;
    um[tmp] = i;
    um2[i] = tmp;
  }

  for (int i = 0; i < m; i++)
  {
    cin >> tmp;
    if (tmp[0] - '0' >= 0 && tmp[0] - '0' <= 9)
    {
      int num = stoi(tmp);
      cout << um2[num] << "\n";
    }
    else
    {
      cout << um[tmp] << "\n";
    }
  }
  return 0;
}