最近写了一段python代码,目的是判断一个数是否是完美数,就是是否可以表达为另一个数据的k次方,即n=m**k,并求出m和k。
import math
def isPP(n):
m=2
res=[]
while m<=round(math.sqrt(n)):
mid=math.log(n,m)
if mid.is_integer():
res.append(m)
res.append(int(round(mid)))
break
m+=1
if len(res)==0:return None
return res
以上代码大部分n是正常工作的,但在n为125,243,216等值时,会返回None,也就是判断其不是完美数。但125是5的3次方,243是3的5次方,216是6的3次方。所以判断问题出在mid.is_integer()这句上。测试一下:
>>> math.log(125,5)
3.0000000000000004
>>> math.log(216,6)
3.0000000000000004
>>> math.log(243,3)
4.999999999999999
>>> math.log(125,5).is_integer()
False
>>> math.log(243,3).is_integer()
False
>>> math.log(216,6).is_integer()
False
就是说浮点数的is_integer()函数是不靠谱的,不能可靠判别是否整数。所以后来的代码改成:
import math
def isPP(n):
m=2
res=[]
while m<=round(math.sqrt(n)):
mid=math.log(n,m)
if abs(mid-round(mid))<0.000000001:
res.append(m)
res.append(int(round(mid)))
break
m+=1
if len(res)==0:return None
return res
这里只能用round,不能用int(),因为用int的话,会截断小数,可能造成mid-int(mid)=1.0的情况。