PTA-Python题目 第2章-4 特殊a串数列求和

简单易懂【PTA-Python题库】第2章-4 特殊a串数列求和

给定两个均不超过9的正整数a和n，要求编写程序求a+aa+aaa++⋯+aa⋯a（n个a）之和。

输入格式：

输入在一行中给出不超过9的正整数a和n。

输出格式：

在一行中按照“s = 对应的和”的格式输出。

输入样例：

2 3

输出样例：

s = 246

思路：

这里我用的是列表推导式，详细请看书P44-46，列表推导式真的可以减少很多不必要的工作，建议好好学习一下。

代码如下：

a, n = input().split()
b = int(n)
s = sum([int(a * i) for i in range(1,b+1)])
print(f's = {s}')