Semi-prime H-numbers(POJ 3292)

Description

形容4n+1的数被划为一个集合，n为自然数，h素数为在该集合中只能被一和自己整除的数，h合数为该集合中能其只能被两个h素数整除的数。读入n，求1~n内有多少个h合数。

Solution

基本数论题。
考虑拓展筛法。
若i为素数，则5i+4i*x也一定为h合数，因为
(5i+4i*x) mod 4=5i mod 4=(5 mod 4)*(i mod 4)=1
并且
（5i+4i*x）=i*(5+4x）

Code

#include <cstdio> 
#include <cstring>
#include <cmath>
#include <algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int n=1000011;
ll m,i,j,cnt;
ll pr[n+5],sum[n+5],p[n+5];
int main(){
    //freopen("1.in","r",stdin);
    //freopen("1.out","w",stdout);
    for(i=5;i<=n;i+=4) {
        if (p[i]==-1) continue;
        pr[++cnt]=i;
        for(j=i*5;j<=n;j+=4*i)
          p[j]=-1;
    }
    fo(i,1,cnt)
      fo(j,1,i)
        if (pr[i]*pr[j]<=n) p[pr[i]*pr[j]]=1;
          else break;
    fo(i,1,n)
      sum[i]=sum[i-1]+max(p[i],0ll);
    scanf("%lld",&m);
    while (m){
          printf("%lld %lld\n",m,sum[m]);
          scanf("%lld",&m);
    }
}