17网易-优雅的点

 

题目描述

 

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

输入描述:

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

输出描述:

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

示例1

输入

25

输出

12


代码

#include<iostream>
#include<algorithm>
#include<vector>
#include<math.h>
using namespace std;

int main() {
    int n, num = 0;
    cin >> n;
    if (n > 0) {
        double b = sqrtf(n), a = sqrtf(n / 2);
        for (int i = ceil(a); i <= int(b); i++)
        {
            if (i == 0) continue;
            double x = sqrtf(n - i*i);
            if (floor(x) == x) {
                if (x*x + i*i == n) {
                    if (x == 0 || x == i)
                        num += 4;
                    else
                        num += 8;
                }
            }
        }
        cout << num << endl;
    }
    else
        cout << 0 << endl;
    return 0;
}