STL_A1100 Mars Numbers (20 分)

https://pintia.cn/problem-sets/994805342720868352/problems/994805367156883456

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

string unitDigit[13]={"tret","jan", "feb", "mar", "apr",
        "may", "jun", "jly", "aug", "sep", "oct", "nov", "dec"};
string higherDigit[13]={"tret","tam", "hel", "maa", "huh", "tou",
        "kes", "hei", "elo", "syy", "lok", "mer", "jou"};

string earthToMars[170];
map<string,int> marsToEarth;

void init() { //打表？
    for(int i=0;i<13;i++) {
        earthToMars[i]=unitDigit[i];
        marsToEarth[unitDigit[i]]=i;
        earthToMars[i*13]=higherDigit[i];
        marsToEarth[higherDigit[i]]=i*13;
    }
    for(int i=1;i<13;i++) {
        for(int j=1;j<13;j++) {
            string str=higherDigit[i]+" "+unitDigit[j]; //注意i,j
            earthToMars[i*13+j]=str;
            marsToEarth[str]=i*13+j;
        }
    }
}

int main() {
    init(); //打表
    int n;
    scanf("%d",&n); 
    getchar(); //吸收空格    
    while(n--) { //用while循环n次
        string str;
        getline(cin,str); //输入字符串
        if(str[0]>='0'&&str[0]<='9') { //如果是数字
            //把数字字符转换为十进制数
            int num=0;
            for(int i=0;i<str.length();i++) {
                num=num*10+(str[i]-'0');
            }
            cout<<earthToMars[num]<<endl;
        }else {
            cout<<marsToEarth[str]<<endl;
        }
    }
    return 0;
}