现有一个Python题需要答疑

特殊a串数列求和
题目内容:
给定两个均不超过9的正整数a和n,要求编写程序a+aa +...+a (n个a)之和
输入格式:
一行中给出不超过9的正整数a和n
输出格式:
一行中按照 "s =和"的格式输出(=前后各有一个英文空格)
输入祥例:
2 3
输出祥例:S= 246

l = list(map(int,input().split(' ')))
a,n = l[0],l[1]
s = 0
for i in range(n):
    s += int(str(a)*(i+1))
print(f'S= {s}')

测试图：

代码如下：

def fn(a,n):
    s=0
    p=""
    for i in range(n):
        p=str(a)*(i+1)
        s=s+int(p)
    return s
a,b=input().split()
s=fn(int(a),int(b))
print(s)

希望对题主有所帮助，可以的话，帮忙点个采纳！