HDU 2866 Special Prime 找规律

题意:称n和m为任意数的p，p为素数 满足等式的p为特殊的素数
求不大于L的特殊的素数的个数
思路:
根据下边的提示很容易想到

1=1^3 2=1^3+1^2
8=2^3 12=2^3+2^2
27=3^3 36=3^3+3^2
然后推下去，随便推导推导就能得到答案了
p=3 * i + 3 * i^2 +1 | i为正整数

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<queue>
#include<stack>
#include<string>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) (x&(-x))
typedef long long LL;
#define maxn 1005
#define Max 1000005
const int inf=(1<<28)-1;
LL ans[maxn];
bool notprime[Max];
int primes[Max];
void get_prime()
{
    notprime[1]=true;
    for(int i=2;i<Max;++i)
    if(!notprime[i])
    {
        primes[++primes[0]]=i;
        for(LL j=(LL)i*i;j<Max;j+=i)
        notprime[j]=true;
    }
}
int main()
{
    int flag;
    get_prime();
    for(int i=1;i<maxn;++i)
    {
        ans[i]=3*i+3*i*i+1;
        flag=i;
        if(ans[i]>=Max) break;
    }
    int k=0;
    for(int i=1;i<flag;++i)
    if(!notprime[ans[i]])
    ans[++k]=ans[i];
    int n;
    while(~scanf("%d",&n))
    {
        int cnt=0;
        for(int i=1;i<=k;++i)
        {
            if(n<ans[i]) break;
            cnt=i;
        }
        if(cnt!=0)
        printf("%d\n",cnt);
        else printf("No Special Prime!\n");
    }
    return 0;
}