HDU 3835 R(N)

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
#include <cmath>
#include <map>
#include <string>
using namespace std;

int n;

int main()
{
    while(~scanf("%d",&n))
    {
        int ans=0;
        for(int i=0; i*i<=n; i++)
        {
            int Nii=sqrt(n-i*i);
            if(Nii*Nii+i*i==n)
            {
                if(i*i==0||n-i*i==0) ans=ans+2;
                else ans=ans+4;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}