PAT A 1100 Mars Numbers (20分)

一、思路

火星文转数字较为简单；
数字专火星文注意：
1、仅数字为0输出"tret"；
2、个位和十位均有输出才输出空格。

注意：以字符为单位输入时注意正确读入换行符。

二、代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <unordered_map>
#include <cctype>
#include <cstdlib>
using namespace std;
int main()
{
    int N;
    scanf("%d\n", &N);
    string mars_1[13] = { "tret", "jan", "feb", "mar", "apr", "may", "jun", "jly", "aug", "sep", "oct", "nov", "dec" };
    string mars_2[13] = { "tret", "tam", "hel", "maa", "huh", "tou", "kes", "hei", "elo", "syy", "lok", "mer", "jou" };
    unordered_map<string, int> earth;
    for(int i = 1; i < 13; ++i)
    {
        earth[mars_1[i]] = i;
        earth[mars_2[i]] = i * 13;
    }
    for(int i = 0, val; i < N; ++i)
    {
        string str;
        getline(cin, str);
        if(isdigit(str[0]))
        {
            val = atoi(str.c_str());
            if(val / 13)
                cout << mars_2[val / 13];
            if(val / 13 && val % 13)
                cout << " ";
            if(val % 13)
                cout << mars_1[val % 13];
            if(!val)
                cout << "tret";
            cout << endl;
        }
        else
        {
            val = 0;
            while(str.size())
            {
                val += earth[str.substr(0, 3)];
                str.erase(0, 4);
            }
            cout << val << endl;
        }
    }
}