HDU 3538 R(N)

HDU 3538 http://acm.hdu.edu.cn/showproblem.php?pid=3835

也是一道卡时间的题目，用正常的思路写，for循环两次，依次寻找A和B是过不了的，则要去改变算法，在找到A的情况下，再去判断B是否满足条件，这样就可以降低复杂度，找到满足平方条件的A和B之后，再用判断A、B是否同时不为0，若A为0或者B为0，则这样的数据要被记录两次，即（0，B）和（0，-B）；若A、B不同时为0，则要记录四次，分别是（A，B）（A，-B）（-A，B）（-A，-B），代码实现如下：

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;

int main()
{
    int n;
    int sum;
    double ex;
    while(scanf("%d",&n)!=EOF)
    {
        sum=0;
        for(int i=0;i<=sqrt(n);i++)
        {
            int t=n-i*i;
            ex=sqrt(t);
            if(ex-(int)ex==0)
            {
                if(i==0||ex==0)
                {
                    sum+=2;
                }
                if(i!=0&&ex!=0)
                {
                    sum+=4;
                }
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}