인프런 커뮤니티 질문&답변

안녕하세요, 1620 나는야 포켓몬 마스터 이다솜 문제관련 질문드립니다.

 

강사님 코드와 거의 비슷하고 atoi 부분만 제가 함수로 작성해서 제출을 했지만 계속 시간초과가 떠서 강사님 코드로 수정해서 돌렸는데도 시간 초과가 뜹니다.. 혹시 아래 코드에서 어디서 문제일까요?

 

 

#include<bits/stdc++.h>
using namespace std;

int n, m;

map<string, int> _map1;
map<int, string> _map2;

int toInt(string s) {
	int ret = 0;
	
	for(auto e : s) {
		ret = ret * 10 + (e - '0');
	}
	return ret;
}

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

	string s;

	cin >> n >> m;
	
	for(int i = 1; i <= n; ++i) {
		cin >> s;
		_map1[s] = i;
		_map2[i] = s;
	}
	
	for(int i = 0; i < m; ++i) {
		cin >> s;
		if('0' <= s[0] && s[0] <= '9') {
			cout << _map2[toInt(s)] << endl;
		} else {
			cout << _map1[s] << endl;
		}
	}

	return 0;
}