算法优化的一个尝试 python

原文：http://www.cnblogs.com/CheeseZH/archive/2012/11/05/2755107.html

题 18

题目：求s=a+aa+aaa+aaaa+aa…a的值，其中a是一个数字。例如2+22+222+2222+22222(此时共有5个数相加)，几个数相加有键盘控制。

原解

“` python 
!/usr/bin/python 
-- coding:utf-8 -- 
def main(): 
basis = int(raw_input(“Input the basis number:”)) 
n = int(raw_input(“Input the longest length of number:”)) 
b = basis 
sum = 0 
for i in range(0,n): 
if i==n-1: 
print “%d “%(basis), 
else: 
print “%d +”%(basis), 
sum+=basis 
basis = basis*10+b 
print ‘= %d’%(sum)

if name == ‘main‘: 
main()

新解

def main():
b = int(input('basis number:'))
n = int(input('longest length:'))
sum = 0
a = 0
for i in range(1,n+1):
    sum += b*10**(i-1)
    a += sum
print(a)

if name == ‘main‘: 
main()