PAT 1044 火星数字 python

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

作者: CHEN, Yue
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB

问题分析:
当个位数为0时不显示，只有当十位个位都为0时显示 tret

代码：

n = int(input())

single = ['', 'jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jly', 'aug', 'sep', 'oct', 'nov', 'dec']
tens = ['', 'tam', 'hel', 'maa', 'huh', 'tou', 'kes', 'hei', 'elo', 'syy', 'lok', 'mer', 'jou']

for i in range(n):
    s = input()
    try:
        s = int(s)
        if s == 0:
            print('tret')
            continue
        a = s % 13  # 个位数
        b = s // 13  # 十位数
        if a == 0 or b == 0:
            rst = tens[b] + single[a]
        else:
            rst = tens[b] + ' ' + single[a]
        print(rst)

    except:
        rst = 0
        s = s.split()
        for j in s:
            if j in single:
                rst += single.index(j)
            elif j in tens:
                rst += tens.index(j) * 13
        print(rst)