今天同学问了我一道python超级素数的题目, 很简单很常规,但是我做的并不好,我打算梳理一下,纠正一下编码习惯。
1.质数模板更改
以前用的是当初大一学VB时的求质数模板,自以为很牛逼,实际上早已落伍且漏洞百出。 以前是这样:
def prime(x):
for i in range(2,x):
if(x%i==0):
break
return (i+1)==x
这种方法以循环变量的值来判断是否是质数,存在的问题是对于012这样的数要单独讨论,不然 i 这个变量不会被赋值,很麻烦,请忘却! 优化后是这样:
def prime(x):
for i in range(2,x):
if(x%i==0):
return False
return True
简单明了,没有用到循环变量,不会有相关的bug,是目前最简单最好理解最好记忆的模板。 倘若要稍微提升下性能,而且也不至于太难记忆,模板可以改成:
def prime(x):
for i in range(2,int(x**0.5)+1):
if(x%i==0):
return False
return True
只是改变了 i 的下限为 根号 x。
2.每n个换行输出
不要再自以为是地用 %n方法了,这样会出现的bug是第一轮换行输出会少一个。 优化为: count从0开始,遍历到 n 则敲回车同时置count 为 0。
3.if语句的优化
不必要的情况下不要用嵌套!还是平行的好,直接编译为机器语言,速度还快。
def prime(x):
for i in range(2,x):
if x % i == 0:
return False
return True
最后的代码如下
count=0
for i in range(2000,10000):
if(prime(i) and prime(int(str(i)[:-1])) and prime(int(str(i)[:-2])) and prime(int(str(i)[:-3]))):
print(i,end=" ")
count+=1
if(count==5):
print()
count=0
print()
|