python编写一个函数mySum(a,n),求以下n项式的和:s=a+aa+aaa+……+aa…a,其中a是1~9的数字,最后一项是n位都是a的数字。样例输入:1,5。输出:12345
时间: 2024-03-21 22:38:30 浏览: 69
计算机Python免费学习第五周作业.pptx
可以使用乘方和等比数列求和的方法来求解,具体实现如下:
```python
def mySum(a, n):
# 计算 aa...a 的值
num = int(str(a) * n)
# 计算 a + aa + aaa + ... + aa...a 的和
s = a * (10**n - 1) // 9 + num * n
return s
```
例如,如果要求 1 + 11 + 111 + 1111 + 11111,则可以这样调用函数:
```python
s = mySum(1, 5)
print(s) # 输出 12345
```
如果要求 2 + 22 + 222 + ... + 222222222,则可以这样调用函数:
```python
s = mySum(2, 9)
print(s) # 输出 246913580
```
这样的实现方法可以在 $O(1)$ 的时间复杂度内完成求和,比使用其他方法更快。
阅读全文