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

 

 

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

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

 

출력을 위한 입력부분에서 숫자로 들어오는지 또는 스트링으로 들어오는지 2가지 경우가 있다.

2가지 경우를 위해 도감도 쪼개주었다.

사실 메모리 초과가 날까봐 살짝 겁은 났는데 돌아가서 기분이 좋다. ^^

 

 

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

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    int regi, qu;
    cin >> regi >> qu;
    map<string, int> dogam_for_str;
    vector<string> dogam_for_index(regi+1);
    
    for(int i = 1; i <= regi; i++){
        string str;
        cin >> str;
        dogam_for_index[i] = str;
        dogam_for_str[str] = i;
    }
    
    for(int i = 1; i <= qu; i++){
        string temp;
        cin >> temp;
        // 숫자인 경우 바로 인덱스로 접근
        if(atoi(temp.c_str()) != 0){
            cout << dogam_for_index[atoi(temp.c_str())] << '\n';
        }else{
            // 문자인 경우
            cout << dogam_for_str[temp] << '\n';
        }
    }
    
    return 0;
}

 

 

atoi인 경우 인자가 숫자가 아닌 경우 0을 리턴한다.

해당 문제의 경우 도감의 인덱스가 1부터 시작하기 때문에 해당 함수를 사용하여 풀 수 있었다.