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

 

백준 1620번_나는야 포켓몬 마스터 이다솜

 

문제는 포켓몬 도감을 받고 싶다는 여러 스토리가 있지만 간단하게 이야기 하자면 ,

아래와 같은 입력과 출력이 나와야 되는 것이다. 

//입력
26 5
Bulbasaur
Ivysaur
Venusaur
Charmander
Charmeleon
Charizard
Squirtle
Wartortle
Blastoise
Caterpie
Metapod
Butterfree
Weedle
Kakuna
Beedrill
Pidgey
Pidgeotto
Pidgeot
Rattata
Raticate
Spearow
Fearow
Ekans
Arbok
Pikachu
Raichu
25
Raichu
3
Pidgey
Kakuna

//출력
Pikachu
26
Venusaur
16
14

 

이 문제는 맵을 사용하고, 또 int 값과 string 값을 확인할 수 있는 것이 핵심같다. 

 

#include <map>
#include <string>
#include <iostream>

using namespace std;

map<string, int> poketmonString;
map<string, string> poketmonInteger;


int main(void) {
    ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
    
	int n, m;
	
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
        string input;
		cin >> input;
		poketmonString[input] = i + 1;
		poketmonInteger[to_string(i + 1)] = input;
	}

	for (int j = 0; j < m; j++) {
        string answer;
		cin >> answer;
		if (poketmonString.count(answer)){
			cout << poketmonString[answer] << '\n';
		}else if(poketmonInteger.count(answer)){
			cout << poketmonInteger[answer] << '\n';
		}
	}
	return 0;
}