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

🥅 문제

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

 

🔍 설계

포켓몬들의 이름이 주어지고, 입력이 포켓몬의 이름이면 번호를, 번호이면 포켓몬의 이름을 출력해야 한다. C++에서 string과 int를 연관지으면서도 속도가 빠른 자료구조에는 트리 기반인 map이 있다. map을 활용하여 포켓몬 도감을 만든다. 

 

key:value가 이름:번호, 번호:이름인 도감을 두 개 만들어서 해결했다. map 클래스에서 find 함수는 map 안에 전달하는 key 값이 없다면, map.end()를 반환한다. 

 

👊 풀이

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
#include <cmath>
using namespace std;	
int n, m;
string name, target;
map<string, int> dogam1;
map<int, string> dogam2;

int main() {
	cin.tie(NULL);
	ios_base::sync_with_stdio(false);
	cin >> n >> m;
	for (int i=1; i<=n; i++){
		cin >> name;
		dogam1.insert({name, i});
		dogam2.insert({i, name});
	}
	for (int i=1; i<=m; i++){
		cin >> target;
		if (dogam1.find(target) == dogam1.end()) {
			cout << dogam2.find(stoi(target))->second << "\n";
		}
		else{
			cout << dogam1.find(target)->second << "\n";
		}
	}
	
	return 0;
}

 

도감을 두 개로 나누어서 해결했지만, 번호도 string 취급하여 하나의 도감에서 해결하는 방법도 있다.

#include <iostream>
#include <vector>
#include <map>
#include <string>
#include <algorithm>
using namespace std;
int n, m;
string s;
map<string, string> dogam;

int main(void){
    ios::sync_with_stdio(false);
	cin.tie(NULL);
    cin >> n >> m;
    for (int i=1;i<=n;i++){
        cin >> s;
        dogam[s] = to_string(i);
        dogam[to_string(i)] = s;
    }
    for (int i=0; i<m; i++){
        cin >> s;
        cout << dogam[s] << "\n";
    }
    
    return 0;
}