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

————————————
呐，最近开始学python，顺便为了期末考试来复习一下代码之类。（完完全全是因为日常的复习生活太过于枯燥了好么！！） 又不想写C，然后来更新一点超简单的python的题这样子，滑稽~
比较简单的题可能就甩一个代码这样？python小白求轻喷orz
————————————

题目描述：

给定两个均不超过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

我的代码：

def fn( a, b ):
	num = a
	sumnum = 0
	while b>1:
		a = a*10 + num
		sumnum += a
		b = b-1
	sumnum += num
	return sumnum