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

 출처 

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

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

 내 풀이 

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

using namespace std;

map<string, string> pokemons;
string pokemon_number[100001];
string pokemon = "";
int N = 0, M = 0;

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> N >> M;
	for (int i = 1; i <= N; i++)
	{
		//형변환 도감번호 넣음 
		string numbers = to_string(i);
		cin >> pokemon;

		pokemons.insert({ pokemon,numbers });	// 포켓몬 도감번호
		pokemon_number[i] = pokemon;
	}

	for (int i = 0; i < M; i++)
	{
		cin >> pokemon;
	
		//번호 입력 했을시 
		if (pokemon[0] >= '1' && pokemon[0] <= '9')
		{
			int number = stoi(pokemon);
			cout << pokemon_number[number] << "\n";
		}
		else cout << pokemons[pokemon] << "\n";
	}
}

 해설

도감 번호를 입력받았을 시 출력할 문자열 배열과 포켓몬을 통해 도감번호를 출력할 맵을 선언한 후 

if문을 통해 입력받은 것에 따라 배열, 맵을 출력해주었다.

 느낀점 

시간 초과 때문에 쪼끔 어려웠다.