弱校胡策 cubicp（数学）

立方数2(cubicp) Time Limit:1000ms Memory Limit:128MB
题目描述
LYK定义了一个数叫“立方数”，若一个数可以被写作是一个正整数的3次方，则这个数就是立方数，例如1,8,27就是最小的3个立方数。
LYK还定义了一个数叫“立方差数”，若一个数可以被写作是两个立方数的差，则这个数就是“立方差数”，例如7(8-1),26(27-1),19(27-8)都是立方差数。
现在给定一个数P，LYK想要知道这个数是不是立方差数。 当然你有可能随机输出一些莫名其妙的东西，因此LYK有T次询问~
这个问题可能太难了…… 因此LYK规定P是个质数！
输入格式(cubicp.in)
第一行一个数T，表示有T组数据。
接下来T行，每行一个数P。
输出格式(cubicp.out)
输出T行，对于每个数如果是立方差数，输出“YES”，否则输出“NO”。
输入样例
5
2
3
5
7
11
输出样例
NO
NO
NO
YES
NO

数据范围
对于30%的数据p<=100。
对于60%的数据p<=10^6。
对于100%的数据p<=10^12，T<=100。
立方差公式 a^3-b^3=(a-b)*(a^2+a*b+b^2)
因p为质数，后面的那一项显然大于前一项 所以前一项为1，后一项为p。即p=a^2+a*b+b^2
a-b=1 –> b=a-1 代换可得 p=3*a^2-3*a+1 之后枚举判断

#include<iostream>#include<cstdio>#include<cstdlib>using namespace std;int main(){    int t;    scanf("%d",&t);    for(int i=1;i<=t;i++)    {        long long p;        scanf("%lld",&p);        for(long long k=1;k;k++)        {            long long now=3*k*k-3*k+1;            if(now==p)            {                puts("YES");                break;            }            if(now>p)            {                puts("NO");                break;            }        }    }    return 0;}