pku 3292 Semi-prime H-numbers 数论素数筛选的变形

http://poj.org/problem?id=3292

H-number 是4*n + 1 (n >= 1) 而 H-prime h 则是只包含因子 1和h的H-number 这里特别注意h是也要是H-number 所以9在这里就是素数了。才开始自己以为从4*n + 1这批数中选出素数然后枚举所有的乘积，然后记录个数。样例肯定不会过了。

思路一样可就是素数筛选时，要选H-prime

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 1000007
using namespace std;

int p[maxn],dp[maxn],f[maxn];
int len;
void init()
{
    int i,j;
    long long tmp;
    memset(f,0,sizeof(f));
    //关键是这里的筛选
    for (i = 1; (4*i + 1) < maxn; ++i)
    {
        int num = 4*i + 1;
        if (!f[num])
        {
           for (j = num*2; j < maxn; j += num) f[j] = 1;
        }
    }
    len = 0;
    for (i = 1; (4*i + 1) < maxn; ++i)
    {
        int num = 4*i + 1;
        if (!f[num]) p[len++] = num;
    }
    memset(dp,0,sizeof(dp));
    for (i = 0; i < len; ++i)
    {
        for (j = i; j < len; ++j)
        {
            tmp = (long long)p[i]*p[j];
            if (tmp <= 1000001) dp[tmp] = 1;
            else break;
        }
    }
    for (i = 1; i < maxn; ++i)
    {
        dp[i] = dp[i - 1] + dp[i];
    }
}

int main()
{
    int n;
    init();
    while (~scanf("%d",&n))
    {
        if (!n) break;
       printf("%d %d\n",n,dp[n]);
    }
    return 0;
}