BNUOJ 1093 YAPTCHA

CEPC 2008

应用Wilson定理，可知当p为素数时，表达式值为1，否则为0。所以筛出素数表，扫一篇就可以了。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int ans[1200000];
#define MAXN 3300000
bool prime[MAXN];
void getPrime()
{
    int i, j;
    prime[0] = prime[1] = false;
    for(i = 2; i < MAXN; i ++)
    {
        prime[i] = true;
    }
    for(i = 2; i * i < MAXN; i ++)
    {
        if(prime[i]) {
            for(j = i * i; j < MAXN; j += i)
            {
                prime[j] = false;
            }
        }
    }
}
void init()
{
    int i;
    getPrime();
    ans[1]=0;
    for (i=2; i<=1000000; i++)
    {
        if (prime[3*i+7] == true)
            ans[i]=ans[i-1]+1;
        else
            ans[i]=ans[i-1];
    }
}
int main()
{
    int prob,i,t,n;
    init();
    scanf("%d",&prob);
    while (prob--)
    {
        scanf("%d",&n);
        printf("%d\n",ans[n]);
    }
}