题目44：找出最小的和与差都是五角数的五角数对

cq_xuke

1. import time
   
2. s=time.clock()
   
3. from itertools import takewhile
   
4. def pentagon(init=None):
   
5.     if init is None:
   
6.         init=1
   
7.     while not ispentagon(init):
   
8.         init+=1
   
9.     n=((1+24*init)**0.5+1)//6
   
10.     while True:
    
11.         yield n*(3*n-1)//2
    
12.         n+=1
    
13. 
    
14. def ispentagon(n):
    
15.     flag=False
    
16.     if int(((1+24*n)**0.5+1)/6)==((1+24*n)**0.5+1)/6:
    
17.         flag=True
    
18.     return flag
    
19. 
    
20. flag=False
    
21. for p1 in pentagon(3):
    
22.     if flag==True:
    
23.         break
    
24.     for p2 in takewhile(lambda x:x<p1,pentagon(1)):
    
25.         if ispentagon(p1+p2) and ispentagon(p1-p2):
    
26.             print(int(p1-p2))
    
27.             flag=True
    
28.             break
    
29. print('Runing time: {0}s'.format(round(time.clock()-s,3)))

复制代码

5482660
Runing time: 3.316s