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

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

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

 

#include <iostream>
#include <map>
#include <string>
using namespace std;
 
int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
 
	int n, m;
	cin >> n >> m;
 
	map<int, string> m1;
	map<string, int> m2;
	string s;
	for(int i = 1; i <= n; i++){
		cin >> s;
		m1.insert(make_pair(i, s));
		m2.insert(make_pair(s, i));
	}
 
	for(int i = 0; i < m; i++){
		cin >> s;
		if('0' <= s[0] && s[0] <= '9'){
			int idx = stoi(s);
			cout << m1[idx] << '\n';
		} else{
			cout << m2[s] << '\n';
		}
	}
	return 0;
}

map 컨테이너는 [ ]로 key 값을 이용해 value 값을 찾을 수 있다. key값이 string인 map과 int인 map 두 개를 만들어서 입력되는 문제가 string일 경우와 int인 경우 각각 다른 map 컨테이너를 검색하도록 구현하였다.