本文实例讲述了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程序设计有所帮助。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:notice#nhooo.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。