python-使用函数求特殊a串数列和

给定两个均不超过9的正整数a和n，要求编写函数fn(a,n) 求a+aa+aaa++⋯+aa⋯aa(n个a）之和，fn须返回的是数列和

函数接口定义：

1 fn(a,n)
2 其中 a 和 n 都是用户传入的参数。 a 的值在[1, 9]范围；n 是[1, 9]区间内的个位数。函数须返回级数和

裁判测试程序样例：

1 /* 请在这里填写答案 */
2
3 a,b=input().split()
4 s=fn(int(a),int(b))
5 print(s)

输入样例：

在这里给出一组输入。例如：

2 3

输出样例：

在这里给出相应的输出。例如：

246

代码：

def fn(a, n):
    sum_ = 0
    for i in range(1, n + 1):
        sum_ += int(str(a) * i)
    return sum_