网易2017秋招编程题-优雅的点

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

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

输入例子:
25

输出例子:
12

#include<stdio.h>
#include<math.h>
int main(){
    int num,r,i,j,count=0;
    scanf("%d",&num);
    r=(int)sqrt(num);
    if(r*r==num){
       for(i=1;i<r;i++){
        for(j=r-i+1;j<r;j++){
            if(i*i+j*j==num){
                count++;
            }
        }
    }
    printf("%d",4*count+4);
    }else{
        for(i=1;i<=r;i++){
            double n=sqrt(num-i*i);
            j=(int)n;
            if(j==n)
                count++;
        }
        printf("%d",4*count);
    }
    return 0;
}