牛客网 21天 2.1禁忌雷炎

赫柏在绝域之门击败鲁卡斯后，从鲁卡斯身上掉落了一本高级技能书，赫柏打开后惊喜地发现这是一个早已失传的上古技能---禁忌雷炎。
该技能每次发动只需扣很少的精神值，而且输出也非常高。
具体魔法描述如下：
把地图抽象为一个二维坐标，技能发动者位于(0,0)位置。以技能发动者为中心，做一个半径为r的圆，满足r^2=S，如果敌人位于这个圆上，且位置为整点坐标，这个敌人将收到该技能的输出伤害。。
例如当S=25时，将有12个敌人受到该技能的输出伤害，如下图所示：

牛客网 21天 2.1禁忌雷炎

更厉害的是，禁忌雷炎可以通过改变魔法输入来控制S的大小，因此数学好的魔法师可以通过该技能攻击到更多的敌人。
赫柏想将这个技能学会并成为自己的主技能，可数学是他的硬伤，所以他请求你为他写一个程序，帮帮他吧，没准他就把禁忌雷炎与你分享了 : )

输入描述:

多组测试数据，请处理到文件结束。

对于每组测试数据，只包含一个整数S。

保证：

1<=S<=2,000,000,000。

输出描述:

输出一个整数，代表受到禁忌雷炎伤害的敌人数量。

示例1

输入

25
3

输出

12
0

c++

#include <bits/stdc++.h>
using namespace std;
int main(){
    int n;
    while(cin>>n){
        int ans=0;
        for(int i=0; i*i<n; i++){
            //原理是判断点到点之间的距离等于半径
            int j = n-i*i;
            int s = sqrt(j);
            if(s*s == j)ans++;
        }
        cout<<4*ans<<endl;
    }
    return 0;
}

python

import math
while True:
    try:
        n = int(raw_input().strip())
        ans = 0
        for i in range(1, int(math.sqrt(n)+1)):
            j = n - i*i
            s = int(math.sqrt(j))
            if s*s==j:
                ans+=1
        print(4*ans)
    except EOFError:
        break

java

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner reader = new Scanner(System.in);
        while(reader.hasNext()){
            int n = reader.nextInt();
            int ans = 0;
            for(int i=1; i*i<=n; i++){
                int j = n - i*i;
                int s = (int)Math.sqrt(j);
                if(s*s == j){
                    ans++;
                }
            }
            System.out.println(ans*4);
        }
    }
}