摘要:面试官:你知道怎么求素数吗?我:求素数?
本文分享自华为云社区《很多人不知道的求素数的正确方法》,原文作者:bigsai 。
前言
现在的面试官,是无数开发者的梦魇,能够吊打面试官的属实不多,因为大部分面试官真的有那么那几下子。但在面试中,我们这些小生存者不能全盘否定只能单点突破—从某个问题上让面试官眼前一亮。这不,今天就来分享来了。
这年头,算法岗内卷不说,开发岗也有点内卷,对开发者要求越来越高了,而面试官也是处心积虑的 “刁难” 面试者,凡是都喜欢由浅入深,凡是都喜欢问个:你知道为什么?你知道原理吗?之类。并且,以前只是大厂面试官喜欢问算法,大厂员工底子好,很多甚至有ACM经验或者系统刷题经验,这很容易理解,但现在一些小公司面试官也是张口闭口 xx算法、xx数据结构你说说看,这不,真的被问到了。
求一个质数
在这么一次的过程,面试官问我算法题我不吃惊,我实现早把十大排序原理、复杂度分析、代码手写实现出来了,也把链表、树的各种操作温习的滚瓜烂熟,不过突然就是很诧异的面试官来了一道求素数问题,我把场景还原一下:
面试官:你知道怎么求素数吗?
我:求素数?
面试官:是的,就是求素数。
我:这很简单啊,判断一个数为素数,那么肯定就没有两个数(除了自身和1)相乘等于它,只需要枚举看看有没有能够被它整除的数就可以了,如果有那么就不是素数,如果没有,那么就是素数。
面试官露出一种失望的表情,说我说的对,但没答到点子上,让我具体说一下。
下面开始开始我的表演:
首先,最笨的方法,判断n是否为素数,就是枚举[2,n-1]之间有没有直接能够被n整除的,如果有,那么返回false这个就不是素数,否则就是素数,代码如下:
boolean isprime(int value){for(int i=2;i<value;i++){if(value%i==0){return false;}}return true;}
这种判断一个素数的时间复杂度为O(n).
但是其实这种太浪费时间了,完全没必要这样,可以优化一下。如果一个数不是质数,那么必定是两个数的乘积,而这两个数通常一个大一个小,并且小的小于等于根号n,大的大于等于根号n,我们只需要枚举小的可能范围,看看是否能够被整除,就可以判断这个数是否为素数啦。例如100=2*50=4*25=5*20=10*10 只需要找2—10这个区间即可。右侧的一定有个对应的不需要管它。
boolean isprime(int value){for(int i=2;i*i<value+1;i++){if(value%i==0){return false;}}return true;}
这里之所以要小于value+1,就是要包含根号的情况,例如 3*3=9.要包含3.这种时间复杂度求单个数是O(sqrt(n))。面试官我给你画张图让你看看其中区别:
说到这里面试官露出欣慰的笑容。
面试官:不错不错,基本点掌握了
我:老哥,其实求素数精髓不在这,这个太低效在很多时候,比如求小于n的所有素数,你看看怎么搞?
面试官:用个数组用第二种方法求O(n*sqrt(n))还行啊。
求多个素数
求多个素数的时候(小于n的素数),上面的方法就很繁琐了,因为有大量重复计算,因为计算某个数的倍数是否为素数的时候出现大量的重复计算,如果这个数比较大那么对空间浪费比较多。
这样,素数筛的概念就被发明和使用。筛的原理是从前往后进行一种递推、过滤排序以来统计素数。
埃拉托斯特尼(Eratosthenes)筛法
我们看一个数如果不是为素数,那么这个数没有数的乘积能为它,那么这样我们可以根据这个思想进行操作啊:
直接从前往后枚举,这个数位置没被标记的肯定就是素数,如果这个数是素数那么将这个数的倍数标记一下(下次遍历到就不需要在计算)。如果不是素数那么就进行下一步。这样数值越大后面计算次数越少,在进行具体操作时候可借助数组进行判断。所以埃氏筛的核心思想就是将素数的倍数确定为合数。
假设刚开始全是素数,2为素数,那么2的倍数均不是素数;然后遍历到3,3的倍数标记一下;下个是5(因为4已经被标记过);一直到n-1为止。具体流程可以看图:
具体代码为:
boolean isprime[];long prime[];void getprime(){prime=new long[100001];//记录第几个primeint index=0;//标记prime当前下标isprime=new boolean [1000001];//判断是否被标记过for(int i=2;i<1000001;i++){if(!isprime[i]){prime[index++]=i;}for(int j=i+i;j<1000000;j=j+i)//他的所有倍数都over{isprime[j]=true;}}}
这种筛的算法复杂度为O(nloglogn);别小瞧多的这个logn,数据量大一个log可能少不少个0,那时间也是十倍百倍甚至更多的差距。
欧拉筛
面试官已经开始点头赞同了,哦哦的叫了起来,可其实还没完。还有个线性筛—欧拉筛。观察上述的埃氏筛,有很多重复的计算,尤其是前面的素数,比如2和3的最小公倍数为6,每3次2的计算就也会遇到是3的倍数,而欧拉筛在埃氏筛的基础上改进,有效的避免了这个重复计算。
具体是何种思路呢?就是埃氏筛是遇到一个质数将它的倍数计算到底,而欧拉筛则是只用它乘以已知晓的素数的乘积进行标记,如果素数能够被整除那就停止往后标记。
在实现上同样也是用两个数组,一个存储真实有效的素数,一个用来作为标记使用。
在遍历到一个数的时候,如果这个数没被标记,那么这个数存在素数的数组中,对应下标加1.不管这个数是不是素数,遍历已知素数将它和该素数的乘积值标记,如果这个素数能够被当前值i整除,那么停止操作进行下一轮。
具体实现的代码为:
boolean isprime[];int prime[];void getprimeoula()// 欧拉筛{prime = new int[100001];// 记录第几个primeint index = 0;isprime = new boolean[1000001];for (int i = 2; i < 1000001; i++) {if (!isprime[i]) {prime[index++] = i;}for (int j = 0; j < index && i * prime[j] <= 100000; j++){//已知素数范围内枚举isprime[i * prime[j]] = true;// 标记乘积if (i % prime[j] == 0)break;}}}
你可能会问为啥if (i % prime[j] == 0)就要break。
如果i%prime[j]==0,那么就说明i=prime[j]*k. k为一个整数。
那么如果进行下一轮的话
i*prime[j+1]=(prime[j]*k)*prime[j+1]=prime[j]*(k*prime[j+1]) 当i=k*prime[j+1]两个位置就产生冲突重复计算啦,所以一旦遇到能够被整除的就停止。
你可以看到这个过程,6只标记12而不标记18,18被9*2标记。详细理解还需要多看看代码想想。过程图就不画啦!欧拉的思路就是离我较近的我给它标记。欧拉筛的时间复杂度为O(n),因为每个数只标记一次。
面试官露出一脸欣赏的表情,说了句不错,下面就是聊聊家常,让我等待下一次面试!
点击关注,第一时间了解华为云新鲜技术~
如果觉得《面试官:你知道怎么求素数吗?》对你有帮助,请点赞、收藏,并留下你的观点哦!