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

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

시간초과에 걸리지 않기 위해서 해시 맵을 사용해야 한다.

 

---

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

using namespace std;

int main()
{
	vector<string> answer;
	vector<string> nameList;
	map<string, int> idxList;
	int M, N;
	cin >> M >> N;

	for (int i = 0; i < M; i++)
	{
		string str;
		cin >> str;

		nameList.push_back(str);
		idxList.insert(make_pair(str, i + 1));
	}

	for (int i = 0; i < N; i++)
	{
		string str;
		cin >> str;

		if (str[0] >= 'A' && str[0] <= 'Z')
			answer.push_back(to_string(idxList.find(str)->second));
		else
			answer.push_back(nameList[stoi(str) - 1]);
	}

	for (int i = 0; i < N; i++)
		cout << answer[i] << '\n';

	return 0;
}

nameList와 idxList 두 개를 사용하여 구현하였다.

 

입력이 들어오면 nameList에는 포켓몬 이름을 저장하고, idxList에는 이름과 번호를 저장해주었다.

추 후 인덱스로 입력이 들어오는 경우는 nameList를 이용해 답을 저장해주었고, 이름으로 들어오는 경우는 find() 를 활용하여 저장하였다.