年龄问题

问题描述：

王二、张三、李四、刘五年龄成一等差数列，他们四人的年龄相加是28，相乘是585。
求以他们的年龄为前4项的等差数列的前15项。

我的代码：

def fun(sum,cj):
for a in range(1,7):
for d in range(1,5):
sum=2*a+3*d
cj=a*(a+d)*(a+2*d)*(a+3*d)
if sum==14 and cj==585:
print "wang er is %d old,the difference is %d" % (a,d)
return a,d
aa,dd=fun(28,585)
for i in range(0,15):
print aa+dd*i,

我的思路：

按照等差数列通项公式，假设王二年龄为a，公差为d，那么满足条件2a+3d=14和

a(a+d)(a+2d)(a+3d)=585，又因为a的范围为1到7，d的范围为1到5，那么就可以采用穷举法求得a和d的值，进而就可以求得该等差数列的前15项了；

示例代码：

def sum(a, k, n):
s = a
for i in xrange(1, n):
s += a + i * k
return s
def mul(a, k, n):
s = a
for i in xrange(1, n):
s *= a + i * k
return s
for a in range(1, 28 / 4):
find = False
k = 1
while True:
t = sum(a, k, 4)
if t >= 28:
if t == 28 and mul(a, k, 4) == 585:
find = True
break
k += 1
if find:
for i in xrange(4):
print a + i * k,

题目出处：http://www.cheemoedu.com/exercise/46