[BZOJ1041]圆上的整点

题意：求圆Ｏ: x^2+y^2=r^2(r>0)上坐标为整点的个数

　　移向项　y^2=r^2-x^2=(r-x)(r+x)

　　设d=gcd(r-x,r+x)得　y^2=(d^2)*(r-x)/d*(r+x)/d

　　设A=(r-x)/d,B=(r+x)/d，得A+B=2*r/d

　　因为A，B为整数，所以d为2*r的因数，所以在区间[1，sqrt(2*r)]

　　又因为y^2和d^2均为完全平方数，AB互质，所以AB均为完全平方数

　　设A=a^2,B=b^2

　　2*r/d=a^2+b^2>=2*a^2>=2*a

　　再在区间[1，sqrt(r/d)]和区间[1，sqrt(d/2)]上枚举a，算出b，判断是否互质即可

#include<bits/stdc++.h>
using namespace std;

typedef long long i64;
i64 gcd(i64 x,i64 y){
    return y?gcd(y,x%y):x;
}
bool check(i64 a,double b){
    i64 bb=(long long)b;
    if(bb!=b)return false;
    i64 A=a*a,B=bb*bb;
    return A!=B&&gcd(A,B)==1;
}
int main(){
    i64 R;
    scanf("%lld",&R);
    i64 lim=sqrt(2*R);
    int ans=0;
    for(i64 d=1;d<=lim;d++){
        if(2*R%d==0){
            i64 up=sqrt(R/d);
            for(i64 a=1;a<=up;a++){
                double b=sqrt(2*R/d-a*a);
                if(check(a,b))ans++;
            }
            if(2*R/d!=d){
                up=sqrt(d/2);
                for(i64 a=1;a<=up;a++){
                    double b=sqrt(d-a*a);
                    if(check(a,b))ans++;
                }
            }
        }
    }
    printf("%d
",ans*4+4);
    return 0;
}