求质数算法

2021-04-05 19:01:58 作者：互联网

每天一个算法～求质数算法

import math
def sieve(size):    
     sieve= [True] * size    
     sieve[0] = False    
     sieve[1] = False    
     for i in range(2, int(math.sqrt(size)) + 1):       
          k= i * 2        
          while k < size:           
              sieve[k] = False           
              k += i    
              return sum(1 for x in sieve if x)
print(sieve(10000000000))

455052511

求解过程

方法1、递归函数

这是一个求质数个数的题不说了
简单做一个递归的优化，每次都用质数筛

@numba.jit()
def cur(size):
    sieve = [True] * size
    sieve[0] = False
    sieve[1] = False
    if size == 2:
        return sieve
    factor = [index for index, val in enumerate(cur(int(math.sqrt(size)+1))) if val]
    for i in factor:
        k = i * 2
        while k < size:
            sieve[k] = False
            k += i
    return sieve

def up(size):
    sieve = cur(size)
    return sum(1 for x in sieve if x)

关键在于@numba.jit()这一行，用了numba的及时解释器，把cur函数重新编译了。在我的mac上大概10分钟不到就可以出结果。很多数值计算在python 里面超级慢，但是用了numba的jit基本可以和c的效率同一数量级。并且写起来很方便，可以尽情地写for循环，方便jit理解，实际运行效率更高。用tensorflow写了神经网络顺便想用python简单处理一下数据，但是又被python的效率限制的童鞋可以尝试一下。我自己实测超级好用。（其实是我不会写c，233

方法2、暴力破解

https://leetcode-cn.com/problems/count-primes/solution/ji-shu-zhi-shu-bao-li-fa-ji-you-hua-shai-fa-ji-you/
直接贴网址了，整体复制过来还要调整合适

方法3、死记硬背

我的智商有限，只能记四位数字，能者多劳，你们加油！！！
在这里插入图片描述

方法4、numba外挂加速

https://zhuanlan.zhihu.com/p/60994299
我还没学完，希望看懂的留言给我