A Cubic number and A Cubic Number HDU - 6216

2018TYUT暑期ACM模拟赛(8)
​​​A Cubic number and A Cubic Number HDU - 6216 ​​​
题意：判断素数p是否为两个立方数的差值。立方数：1 8 27 64 125…..
思路：立方数之差 x^3-y^3=(x-y)(x^2+y^2+xy) 由于p是素数，只能表示1*p 所以 x-y=1。
把x=y+1带入第二个因子。3(y^2+y)+1=p。所以只要判断p满足这个关系式即可。(p-1)%3==0。y(y+1)=p/3。从 i=根号p 开始搜索，直到i*i+i>p

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int main()
{
    int t;
    long long p;
    scanf("%d",&t);
    while(t--)
    {
        cin>>p;
        int flag=0;
        p-=1;
        if(p%3==0)
        {
            p/=3;
            for(long long i=sqrt(p);i*(i+1)<=p;i++)
                if(i*(i+1)==p)
                {
                    printf("YES\n");
                    flag=1;
                    break;
                }
            if(!flag)
            printf("NO\n");
        }
        else printf("NO\n");
    }
    return 0;
}