第6章函数-1 使用函数求特殊a串数列和 (10分)

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

函数接口定义：

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

思考 :

因为用C语言写过这个题目，思路还是清楚的。但是Python使用的不熟练，写的时间过长了。

答案：

def fn(a, b):
    res = sum = 0
    list = []
    for x in range(1, b + 1):   #[1, 2, ..., b]
        res = res * 10 + 1      
        list.append(res)        #[1, 11, 111, ...]
    for i in list:
        sum = sum + a * i       #[a, aa, aaa, ...]
    return sum  #必须要有返回值

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