A Cubic number and A Cubic Number HDU - 6216

$a^3-b^3=p$ $a^3-b^3=(a-b)(a^2+ab+b^2)$,$p$ 是素数
a-b=1 $a^2+ab+b^2=p$
代入得到 $3b^2+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;
}