python编写一个函数mySum(a,n),求以下n项式的和:s=a+aa+aaa+……+aa…a,其中a是1~9的数字,最后一项是n位都是a的数字。
时间: 2024-03-21 18:38:28 浏览: 224
python 求s=a+aa+aaa+aaaa+aa...a的值,其中a是一个数字 (示例)
5星 · 资源好评率100%
可以使用乘方和等比数列求和的方法来优化算法,具体实现如下:
```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)$ 的时间复杂度内完成求和,比使用等差数列求和的方法更快。
阅读全文