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

문제

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

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

문제풀이

1부터 N까지 포켓몬 이름을 입력받고 도감 번호를 입력받으면 해당 포켓몬 이름을 출력, 포켓몬 이름을 입력받으면 도감 번호를 출력하면 된다.

포켓몬 이름을 입력받으면 pokemonS에, 도감 번호를 입력받으면 pokemonI에 저장한다.

도감 입력이 모두 끝났으면 입력에 따른 출력을 하면 된다.

입력이 숫자라면 string을 int로 형 변환하여 pokemonI[번호]의 값을,

포켓몬 이름이라면 pokemon[포켓몬 이름]의 값을 출력한다.

 

코드

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

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

	int N(0), M(0), idx(0);
	string input("");

	cin >> N >> M;

	unordered_map<string, int>pokemonS;
	unordered_map<int, string>pokemonI;

	for (idx = 1; idx <= N; ++idx) {
		cin >> input;
		pokemonS.insert({ input,idx });
		pokemonI.insert({ idx, input });
	}

	while (M--) {
		cin >> input;
		if (input[0] >= '0' && input[0] <= '9')
			cout << pokemonI[stoi(input)];

		else cout << pokemonS[input];
		cout << '\n';
	}
}