版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/becky_w/article/details/83239767
,
是素数
a-b=1
代入得到
算出所有可能的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;
}