蓝桥杯python：天干地支

题目：

程序说明：
先输入一个整型数，判断它是否小于9999，接着判断它与2020的关系，进行后续的程序操作。其实之前我并没有想着列出Tg，Dz两个列表来进行计算，就想着当数据小于2020时，接着找到一定的规律，输出。最后显然我很菜，没有找到所谓的规律，然后因为看了另外一位大佬的操作后，好吧，最后还是列出了由gengzi倒着数过去的列表，ok最后运行出来了。

全部代码：

n=int(input())
tg=['jia','yi','bing','ding','wu','ji','geng','xin','ren','gui']
dz=['zi','chou','yin','mao','chen','si','wu','wei','shen','you','xu','hai']
Tg=['geng','ji','wu','ding','bing','yi','jia','gui','ren','xin']
Dz=['zi','hai','xu','you','shen','wei','wu','si','chen','mao','yin','chou']
if n<9999:
    if n==2020:
        print(tg[6]+dz[0])
    elif n>2020:
        a=n-2020
        print(tg[(6-a)%10]+dz[(0+a)%12])
    else:
        b=2020-n
        print(tg[(b%10)]+dz[(b%12)])