【bzoj 1002】[FJOI2007]轮状病毒 python水过

不想写高精度

a=[0]*105
b=[0]*105
a[1]=1
a[2]=4
b[1]=1
b[2]=3
for i in range(3,100):
    a[i]=3*a[i-1]-a[i-2]
    b[i]=3*b[i-1]-b[i-2]
n=int(input())
if(n%2): print ((int(a[int((n+1)/2)]))**2)
else: print (((b[int(n/2)])**2)*5)