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

🔗 문제 링크

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

✍🏻 문제 설명

안녕? 내 이름은 이다솜

💁🏻‍♀️ 내 풀이

스토리가 너무 길다..😱 다 건너뛰고 마지막 오박사님 말씀만 보면 될 거 같다!

 

오박사 : 그럼 다솜아 이제 진정한 포켓몬 마스터가 되기 위해 도감을 완성시키도록 하여라. 일단 네가 현재 가지고 있는 포켓몬 도감에서 포켓몬의 이름을 보면 포켓몬의 번호를 말하거나, 포켓몬의 번호를 보면 포켓몬의 이름을 말하는 연습을 하도록 하여라. 나의 시험을 통과하면, 내가 새로 만든 도감을 주도록 하겠네.

 

포켓몬 매핑 문제라고 이해하면 된다!

 

예시 : 26개가 들어오고, 맞춰야할 5개가 들어옴

 

M개 문제 중에서 번호를 입력받으면 해당 포켓몬의 이름을 출력하고, 이름을 입력받으면 해당 포켓몬의 번호를 출력하는 문제인데, 배열로 풀면 시간초과가 난다.

 

그래서 map을 사용해서 푸는데, string -> int , int -> string 맵을 2개 만들어 줘야 한다.  

🤖 코드

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

int n, m;
string s;
map<string, int> map1;
map<int, string> map2;

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

	for (int i = 1; i <= n; i++)
	{
		cin >> s;
		map1[s] = i;
		map2[i] = s;
	}

	for (int i = 1; i <= m; i++)
	{
		cin >> s;
		if (atoi(s.c_str()) == 0)
		{
			cout << map1[s] << "\n";
		}
		else
		{
			cout << map2[atoi(s.c_str())] << "\n";
		}
	}

	return 0;
}

 

그리고 처음에 시간 초과 났었는데, 아래 두개 추가해주니 맞췄

ios::sync_with_stdio(0);
cin.tie(0);

 

위의 코드가 궁금하다면 아래 블로그 참고

https://seize-the-day9.tistory.com/24

[C++] C++ 입출력 속도를 더 빠르게 처리하는 방법