python求解水仙花数的方法

本文实例讲述了python求解水仙花数的方法。分享给大家供大家参考。具体如下：

一个N位的十进制正整数，如果它的每个位上的数字的N次方的和等于这个数本身，则称其为花朵数。

#!/usr/bin/python
def get_flower(n, ofile):
D_pow=[pow(i,n) for i in range(0,10)]
V_min=1*pow(10,n-1)
V_max=sum((9*pow(10,x) for x in range(0,n)))
T_count=0
print D_pow, V_max, V_min
nums=[1]+[0]*(n-1)
print 'Start:', nums
idx=n-1
tmp_l=[0]*10
while True:
 nums[idx]+=1
 if nums[idx]<10:
  j=idx+1
  while j<n:
   nums[j]=nums[idx] # reset
   j+=1
  v=sum((D_pow[x] for x in nums))
  if v<=V_max and v>=V_min:
   T_count+=1
   #test if is flower
   #print 'do test:', ''.join(map(str,nums))
   k=0
   while k<10:
    tmp_l[k]=0
    k+=1
   N=0
   for k in nums:
    tmp_l[k]+=1
    N+=1
   while N>0:
    p=v％10
    if tmp_l[p]>0:
     tmp_l[p]-=1
     N-=1
    else:
     break
    v/=10
   if N==0:
    print >>ofile, 'hit', sum((D_pow[x] for x in nums))
  idx=n-1
 elif idx==0:
  print 'done'
  break
 else:
  idx-=1
print 't_count', T_count
if __name__ == '__main__':
with file('./f.txt', 'wb') as o:
 get_flower(21, o)
 #get_flower(3, o)

希望本文所述对大家的Python程序设计有所帮助。