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

문제

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

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

 

접근 방법

map을 이용하면 간단하게 해결할 수 있다.

map에는 포켓몬 이름과 번호를 한 쌍으로 저장하고, 배열에는 번호 순서에 따라 포켓몬 이름을 저장했다.

입력이 이름으로 주어질 경우는 map의 find함수를 통해 번호를 알아내고

입력이 번호로 주어질 경우는 배열을 인덱스로 접근하여 이름을 알아내었다.

주어진 입력이 문자인지 숫자인지는 isdigit 함수를 통해 확인하였다.

 

코드

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

int n, m;
string str;
map<string, int> num;
string name[100001];

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

	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		cin >> str;
		num.insert({ str, i });
		name[i] = str;
	}
	for (int i = 1; i <= m; i++) {
		cin >> str;
		if (isdigit(str[0])) {
			cout << name[stoi(str)] << '\n';
		}
		else {
			cout << num.find(str)->second << '\n';
		}
	}
}

처음에 시간초과가 나와서 ios::sync_with_stdio(0), cin.tie(0)를 추가해 주었더니 해결되었다.

입력이 많다 보니 해주지 않으면 시간이 오버되는 것 같았다.