题目:求1-n的素数个数,n很大。这里介绍一个模板和一点个人理解。
首先要运用动态规划的思想,设f(i)为[1,n/i]的素数个数,设g(i)为[1,i]的素数个数,显然f(1)即为所求。对于任意1-n的合数,其最小质因子不会超过sqrt(n)(这里设为m),我们要将最小质因子为2-m的所有的合数去掉。考虑f(j),即[1,n/j]的素数和,考虑添加一个最小质因子2之后的更新,将(n/j)分为i份,重复的为f(i*j)-g(i-1)即[1,n/i/j]的部分。即
f[j]-=f[i*j]-g[i-1]
*这里可能比较难懂,不妨先假设i=2,j=1,n=20
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
我们要筛掉1-20里所有的2的合数观察f(2),即1-10的区间,可以发现20 可以被10*2筛掉 18可以被9*2筛掉。。。。4可以被2*2筛掉,但由于f(i)包含(i*j,所以i之前的素数都减重了,所以有减去g(i-1)
经过类似的推理,有方程f(j)-=f(i*j)-g[i-1]。
显然:f(i*j)和g(n/i/j)是相同的(定义),当i*j>m时,可以用g(n/i/j)替代,这样可以保证f,g的空间复杂度都为m。
上代码。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=340000;
ll f[N],g[N];
ll n;
void init(){
ll i,j,m;
for(m=1;m*m<=n;m++) f[m]=n/m-1;
for(i=1;i<=m;i++) g[i]=i-1;
for(i=2;i<=m;i++){//m=sqrt(n)+1
if(g[i]==g[i-1]) continue;
for(j=1;j<=min(m-1,n/i/i);j++){//n/i/j>i-1
if(i*j<m) f[j]-=f[i*j]-g[i-1];
else f[j]-=g[n/i/j]-g[i-1];
}
for(j=m;j>=i*i;j--) g[j]-=g[j/i]-g[i-1];//j/i>i-1
}
}
int main(){
while(~scanf("%lld",&n)){
init();
printf("%lld\n",f[1]);
}
return 0;
}