PAT 甲级 1100 Mars Numbers

PAT 甲级 1100 Mars Numbers

#include <bits/stdc++.h>
using namespace std;
vector<string> mars_low = { "tret", "jan", "feb", "mar", "apr", "may", "jun", "jly",
                           "aug", "sep", "oct", "nov", "dec" };
vector<string> mars_high = { "tret", "tam", "hel", "maa", "huh", "tou", "kes", "hei",
                            "elo", "syy", "lok", "mer", "jou" };
map<string, string> to_mars;
map<string, string> to_earth;
void Init()
{
    for (int i = 0; i < 169; ++i) {
        int high = i / 13;
        int low = i % 13;
        if (high == 0)
            to_mars[to_string(i)] = mars_low[i];
        else if (low == 0)
            to_mars[to_string(i)] = mars_high[high];
        else
            to_mars[to_string(i)] = mars_high[high] + " " + mars_low[low];
    }
    for (auto item : to_mars) {
        to_earth[item.second] = item.first;
    }
}

int main()
{
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
#endif
    Init();
    int N; cin >> N; string s; cin.get();
    while (N--) {
        getline(cin, s);
        isdigit(s[0]) ? (cout << to_mars[s] << endl) : (cout << to_earth[s] << endl);
    }
}