6-3 使用函数求特殊a串数列和 的python代码

def special_sum(n: int, a: int) -> int:
"""
计算特殊a串数列的前n项和

:param n: 数列项数
:param a: 数列中的特殊数
:return: 数列前n项和
"""
def special_term(k: int) -> int:
"""
计算数列中第k项的值

:param k: 数列项数
:return: 数列中第k项的值
"""
if k == 1:
return a
else:
return int(str(special_term(k - 1)) + str(a) * k)

# 计算数列前n项和
s = 0
for i in range(1, n + 1):
s += special_term(i)

return s