校招笔试——优雅的点

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

复制

25

输出

复制

12

#include<iostream>
#include<vector>
using namespace std;

int main(){
    int rr;
    cin >> rr;
    vector<int> points;
    for (int i=0; i*i<=rr; i++){
        points.push_back(i*i);
    }
    int res = 0;
    for (int i=0; i < points.size()-1; i++){
        for (int j=points.size()-1; j>=i; j--){
            if (points[i] + points[j] == rr){
                if (i!=0 && j!=0 && i!=j)
                    res += 8;
                else
                    res += 4;
                break;
            }
            if (points[i] + points[j] < rr)
                break;
        }
    }
    cout << res;
    return 0;
}