백준 1620 (c++) 나는야 포켓몬 마스터 이다솜

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

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

// 회사에 있는 사람

#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include <cstring>
using namespace	std;

int	main()
{
	ios::sync_with_stdio(0),cin.tie(0);

	unordered_map<string, int>	map;
	int	size, n;
	string	input;
	vector<string>	dogam;

	cin >> size >> n;
	dogam.push_back("mutz"); // 0번 더미 추가
	for (int i = 1; i <= size; i++)
	{
		cin >> input;
		map[input] = i;
		dogam.push_back(input);
	}
	while (n--)
	{
		cin >> input;
		if (isdigit(input[0])) // 숫자면
			cout << dogam[stoi(input)] << '\n';
		else
			cout << map[input] << '\n';
	}
}