A Cubic number and A Cubic Number HDU - 6216

$a^3-b^3=p$a3−b3=p $a^3-b^3=(a-b)(a^2+ab+b^2)$a3−b3=(a−b)(a2+ab+b2), $p$p 是素数
a-b=1 $a^2+ab+b^2=p$a2+ab+b2=p
代入得到 $3b^2+3b+1=p$3b2+3b+1=p
算出所有可能的p 然后二分就好了

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<ll>v;
int main()
{
    for(ll i=0;i<1000000;i++)
    {
        ll k = i*i*3+3*i+1;
        v.push_back(k);
    }
    int T;
    scanf("%d",&T);
    while(T--)
    {
        ll p;
        scanf("%lld",&p);
        if(binary_search(v.begin(),v.end(),p)) puts("YES");
        else puts("NO");
    }
    return 0;
}