优雅的点

优雅的点

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

思路比较直观，属于简单题，代码如下

import math

n = int(raw_input())
a = math.sqrt(n)
sum = 0
for i in range(int(a) + 1):
    c = n - i**2
    j = int(math.sqrt(c))
    if j**2 == c:
        if(i == 0 or j == 0):
            sum += 2
        else:
            sum += 4

print sum