나는야 포켓몬 마스터 이다솜 (feat. 백준, 1620번)

나는야 포켓몬 마스터 이다솜
https://www.acmicpc.net/problem/1620

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

반응형

// woohyeon.kim
// https://www.acmicpc.net/problem/1620
#include <iostream>
#include <string>
#include <map>
using namespace std;
int main() {
	ios::sync_with_stdio(false), cin.tie(NULL);
	register int N, M;
	string tmp;
	map<int, string> ans1;
	map<string, int> ans2;
	cin >> N >> M;
	for (register int i = 1; i <= N; ++i)
		cin >> ans1[i], ans2[ans1[i]] = i;
	for (register int i = 0; i < M; ++i) {
		cin >> tmp;
		if ('0' <= tmp.at(0) && tmp.at(0) <= '9')	// 입력이 숫자인 경우
			cout << ans1[stoi(tmp)];
		else						// 입력이 문자열인 경우
			cout << ans2[tmp];
		cout << "\n";
	}
	return 0;
}
// *&)*@*

문자열을 key로 갖는 map과 정수형을 key로 갖는 map을 이용하여 입력된 값이 문자열인 경우 순번을, 정수형인 경우 문자열을 출력합니다.