蓝桥杯 Python组决赛 天干地支

【问题】
古代中国使用天干地支来记录当前的年份。天干一共有十个，分别为：甲（jiǎ）、乙（yǐ）、丙（bǐng）、丁（dīng）、戊（wù）、己（jǐ）、庚（gēng）、辛（xīn）、壬（rén）、癸（guǐ）。
地支一共有十二个，分别为：子（zǐ）、丑（chǒu）、寅（yín）、卯（mǎo）、辰（chén）、巳（sì）、午（wǔ）、未（wèi）、申（shēn）、酉（yǒu）、戌（xū）、亥（hài）。将天干和地支连起来，就组成了一个天干地支的年份，例如：甲子。2020 年是庚子年。每过一年，天干和地支都会移动到下一个。例如 2021 年是辛丑年。每过 60 年，天干会循环 6 轮，地支会循环 5 轮，所以天干地支纪年每 60年轮回一次。例如 1900 年，1960 年，2020 年都是庚子年。给定一个公元纪年的年份，请输出这一年的天干地支年份。
【输入】
输入一行包含一个正整数，表示公元年份。
【输出】
输出一个拼音，表示天干地支的年份，天干和地支都用小写拼音表示（不表示声调），之间不要加入任何多余的字符。
【样例输入】
2020
【样例输出】
gengzi
【评测用例规模与约定】
对于所有评测用例，输入的公元年份为不超过 9999 的正整数。

tiangan = ["jia","yi","bing","ding","wu","ji","geng","xin","ren","gui"]
dizhi = ["zi","chou","yin","mao","chen","si","wu","wei","shen","you","xu","hai"]
year = int(input())
cha = year-2020
tg=""
dz=""
if cha >0:
    if cha<4:
        tg=tiangan[6+cha]
    elif cha>=4:
        tg=tiangan[cha%10-4]
    dz=dizhi[cha%12]
elif cha <0:
    if cha >-6:
        tg=tiangan[6+cha]
    elif cha <=-6:
        tg=tiangan[-((abs(cha))%10)+6]
    dz =dizhi[-abs(cha)%12] 
else:
    print("gengzi")
print(tg+dz)