PAT 乙级 1044 火星数字 （20 分）

1044 火星数字 （20 分）

火星人是以 13 进制计数的：

• 地球人的 0 被火星人称为 tret。
• 地球人数字 1 到 12 的火星文分别为：jan, feb, mar, apr, may, jun, jly, aug, sep, oct, nov, dec。
• 火星人将进位以后的 12 个高位数字分别称为：tam, hel, maa, huh, tou, kes, hei, elo, syy, lok, mer, jou。

例如地球人的数字 29 翻译成火星文就是 hel mar；而火星文 elo nov 对应地球数字 115。为了方便交流，请你编写程序实现地球和火星数字之间的互译。

输入格式：

输入第一行给出一个正整数 N（<100），随后 N 行，每行给出一个 [0, 169) 区间内的数字 —— 或者是地球文，或者是火星文。

输出格式：

对应输入的每一行，在一行中输出翻译后的另一种语言的数字。

输入样例：

4
29
5
elo nov
tam

输出样例：

hel mar
may
115
13


注意：13的倍数情况要进行额外讨论

#include<iostream>
#include<string>
#include<cstdlib>

using namespace std;

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

void earthToMars(string& str)
{
    int num = atoi(str.c_str());

    if (num / 13 > 0 && num % 13 != 0)
        cout << str2[num / 13 - 1] << " " << str1[num % 13] << endl;
    else if (num / 13 > 0 && num % 13 == 0)
        cout << str2[num / 13 - 1] << endl;
    else
        cout << str1[num % 13] << endl;
}

void marsToEarth(string &str)
{
    int sum = 0;

    if (str.size() == 3)
    {
        for (int i = 0; i<13; ++i)
        {
            if (str1[i] == str)
            {
                cout << i << endl;
                return;
            }
        }

        for (int i = 0; i<13; ++i)
        {
            if (str2[i] == str)
            {
                cout << (i+1)*13 << endl;
                return;
            }
        }
    }
    else
    {
        string strLeft = str.substr(0, 3);
        string strRight = str.substr(4);

        for (int i = 0; i<13; ++i)
            if (str2[i] == strLeft)
                sum += (i+1) * 13;

        for (int i = 0; i<13; ++i)
            if (str1[i] == strRight)
                sum += i;

        cout << sum << endl;
    }
}

int main()
{
    int N;
    string str;

    cin >> N;
    getchar();

    for (int i = 0; i<N; ++i)
    {
        getline(cin, str);

        if (str[0] >= 48 && str[0] <= 57)
            earthToMars(str);
        else
            marsToEarth(str);
    }
}