昨天复习了快速幂的原理及做法,今天复习筛法。
今天会提及两种筛质数打方法——埃氏筛和欧拉筛。
https://www.luogu.com.cn/problem/P3383https://www.luogu.com.cn/problem/P3383
题目背景
本题已更新,从判断素数改为了查询第 𝑘 小的素数
提示:如果你使用 cin
来读入,建议使用 std::ios::sync_with_stdio(0)
来加速。
题目描述
如题,给定一个范围 𝑛,有 𝑞 个询问,每次输出第 𝑘 小的素数。
输入格式
第一行包含两个正整数 𝑛,𝑞,分别表示查询的范围和查询的个数。
接下来 𝑞 行每行一个正整数 𝑘,表示查询第 𝑘k 小的素数。
输出格式
输出 𝑞 行,每行一个正整数表示答案。
输入输出样例
输入 #1
100 5 1 2 3 4 5
输出 #1
2 3 5 7 11
先讲讲埃氏筛,通过获取质数,划去合数的方法进行筛选,从2开始,划去2的倍数,4,6,8,10以此类推,而没有被划去的就是素数,再对下一个素数进行相同的操作。总体还是非常简单的。
附上埃氏筛的代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e8 + 10;
int n, m, x, cnt = 0;
int prim[N], vis[N];
void Eratosthense(int n)
{
for(LL i = 2; i <= n; i ++)
{
if(!vis[i])
{
prim[++ cnt] = i;
for(LL j = i * i; j <= n; j += i)
vis[j] = 1;
}
}
}
int main()
{
cin >> n >> m;
Eratosthense(n);
while(m --)
{
cin >> x;
cout << prim[x] << endl;
}
}
但是埃氏筛的效率其实相比之下没有欧拉筛高,用这串代码提交会有三个TLE。接下来我们看看欧拉筛,看看究竟效率高在了哪里。
欧拉筛,相同的,也是从小到大枚举每一个数,划去合数,但它就高明在每一个合数都只会被最小的因数划去。因此只需要被划去一次,如15,如果用埃氏筛会被3,5无差别划去,而欧拉筛只需要被3划去一次,那么如何实现呢?接下来我们讲欧拉筛的两个优化。
优化一:越界中断,我们看到,在埃氏筛中,我们划去合数的范围是从j到j的二次方,而我们题目的要求往往不需要那么多数,所以欧拉筛进行了如下优化:
for(int j = 1; prim[j] * i * 1LL <= n; j ++)
这一串条件代码是枚举已经记录的素数,如果当前素数与之前的某一位素数的乘积超过了我们想要的范围,则没有必要继续下去,就退出循环。
优化二:整除中断,上面我们提到欧拉筛中的合数只被最小质因数划去,就是用来整除判断中断:
if(i % prim[j] == 0)
break;
如果当前合数已经会被之前更小的质数整除,我们便不必再划去一次,以节省时间
因此,欧拉筛的完整代码如下:
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e8 + 1;
const int M = 1e6 + 10;
int n, m, cnt = 0;
int vis[N], prim[M];
void get_prim(int n)
{
for(int i = 2; i <= n; i ++)
{
if(!vis[i])
prim[++ cnt] = i;
for(int j = 1; prim[j] * i * 1LL <= n; j ++)
{
vis[i * prim[j]] = 1;
if(i % prim[j] == 0)
break;
}
}
}
int main()
{
scanf("%d%d",&n,&m);
get_prim(n);
while(m --)
{
int x;
scanf("%d",&x);
printf("%d\n",prim[x]);
}
}
这里再提两嘴,第一嘴——scanf输入确实要比cin输入快,有时如果cin不行可以换scanf试试(别问我怎么知道的)。第二嘴,无论是埃氏筛还是欧拉筛,前者算i*i,后者算prim[j]*i时可能会爆int,所以记得用long long型(别问我怎么知道的,too)。