PTA Python 6-1 sdut-使用函数求a+aa+aaa++⋯+aa.....aaa(n个a）之和

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

函数接口定义：

def fn(a,n):
    

其中， a 和 n 都是传入的参数。 a 的值在[1, 9]范围；n 是[1, 9]区间内的个位数。函数返回数列之和。

裁判测试程序样例：

/* 请在这里填写答案 */

a,b=input().split()
s=fn(int(a),int(b))
print(s)

输入样例：

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

2 3

输出样例：

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

246

答案:

 

def fn(a1, n):
    sum_my = 0
    a2 = a1
    for x in range(n):
        sum_my += a1
        a1 = a1 * 10 + a2
    return sum_my