牛客练习-优雅的点

小易有一个圆心在坐标原点的圆，小易知道圆的半径的平方。小易认为在圆上的点而且横纵坐标都是整数的点是优雅的，小易现在想寻找一个算法计算出优雅的点的个数，请你来帮帮他。
例如：半径的平方如果为25
优雅的点就有：(+/-3, +/-4), (+/-4, +/-3), (0, +/-5) (+/-5, 0)，一共12个点。 

输入描述:

输入为一个整数，即为圆半径的平方,范围在32位int范围内。

 

输出描述:

输出为一个整数，即为优雅的点的个数

示例1

输入

25

输出

12

#include<iostream>
#include<cmath>
using namespace std;
int cmp(int x,int n)
{
    int s = n-x*x;
    int m = (int)sqrt(s);
    if(m*m==s)return 1;
    return 0;
}
int main()
{
    int n;
    cin>>n;
    int len = (int)sqrt(n);
    int cnt = 0;
    for(int i = 1; i <=len; i++)
    {
        if(cmp(i,n))cnt++;
    }
    cout<<cnt*4<<endl;

    return 0;
}