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

#include <iostream>
#include <string>
#include <map>

using namespace std;

string input;
map<string, int> name;
map<int, string> num;

int main()
{
	ios::ios_base::sync_with_stdio(false);
	cin.tie(0);
	int n, m;
	cin >> n >> m;
	for (int i = 1; i <= n; i++)
	{
		cin >> input;
		name[input] = i;
		num[i] = input;
	}
	for (int i = 0; i < m; i++)
	{
		cin >> input;
		if (name[input] != NULL)
			cout << name[input] << '\n';
		else cout << num[stoi(input)] << '\n';
	}
}