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

🖨 입출력

---

• 입력

  • 첫째줄: 도감에 수록된 포켓몬 개수(N), 다솜이가 맞춰야 하는 문제 개수(M)

  • 둘째줄: 1~N번까지의 포켓몬 한 줄씩

  • 셋째줄: 맞춰야 하는 문제 M개 한 줄씩

  • 포켓몬 이름 최대 길이: 20

• 출력

  • 문제가 알파벳 -> 포켓몬 번호로 응답

  • 문제가 포켓몬 번호 -> 포켓몬 번호에 해당하는 문자로 응답

  • M개의 줄에 차례로 문제에 대한 답을 출력

 

 

💻 C++ 제출코드

---

#include <bits/stdc++.h>

using namespace std;

int main()
{
    int N, M; //N: 포켓몬 개수, M: 맞춰야 하는 문제 개수


    scanf("%d %d", &N, &M);

    map<string, int> poketmon1;
    map<int, string> poketmon2;
    vector<string> result;

    for(int i=1; i<=N; i++){
        char input[21]; //포켓몬 이름 최대 길이 20
        scanf("%s", &input);

        poketmon1.insert({input, i});
        poketmon2.insert({i, input});
    }

    for(int i=0; i < M; i++){
        char s[21]; //포켓몬 이름 최대 길이 20

        scanf("%s", &s); 

        if(atoi(s) != 0){ // if integer, char * -> int         
            string result = poketmon2.at(atoi(s));
            char *a = &*result.begin();
            printf("%s\n", a);
        }
        else{ //if string
            string str1(s);
            const char *a = to_string(poketmon1[str1]).c_str();
            printf("%s\n", a);           
        }     
    } 
}

 

 

📝 문제풀이 과정

---

• cout이나 cin은 시간이 오래걸려 시간초과가 발생해서, 이 cout과 cin은 사용하지 않고 구현했다.

• atoi(s) == 0이면 문자열이고, 아니면 숫자인 것으로 문자열, 숫자를 판별했다.

• 숫자가 들어오면 atoi(s)인 key로 poketmon2 맵에서 찾고, char *으로 변환해서 scanf로 출력했다.

• 문자열이 들어오면 문자열 그대로 key로 value를 가져와서 int를 to_string()으로 string 형으로 변환 후, c.str()으로 string을 char * 형으로 변환해서 출력했다.

• 포켓몬 이름의 최대길이가 20이므로, null 포함 21 길이의 문자 배열에 입력 받았다.

 

 

 

👏 결과

---

• 숫자와 문자열로 입력이 들어올 때 처리하기 위한 string, int, char * 간의 변환에 대해 생각해볼 수 있었다.