Python 实现特殊数列求和算法

Python 实现特殊数列求和算法

给定两个均不超过9的正整数，较小的数作为'a'，较大的数作为'n'，要求编写程序求 'a' + 'aa' + 'aaa' + ... + 'aaa...aa'(n个'a') 之和。

例如：'a' = 2, 'n' = 5 时，2 + 22 + 222 + 2222 + 22222 = 24690，输出结果 24690

输入格式

同行输入两个正整数 'a'，'n' 表示两个整数，题目保证：'a' 不大于 9，'n' 不大于 10

输出格式

输出一个正整数 'm'，表示特殊数列之和。

样例

输入样例 2 5

输出样例 24690

思路

根据题意直接模拟即可，每次累加一个数，这个数是上一个数乘以 10 再加上 'a'。

代码如下：

a, n = map(int, input().split())
sum = 0
num = a
for i in range(n):
    sum += num
    num = num * 10 + a
print(sum)