bzoj1041: [HAOI2008]圆上的整点

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=1041

题解

　　1A二连击。虽然是抄的题解
　　显然

$$y^2=(R-x)(R+x)$$
　　令 $d=gcd(R-x,R+x),A=\frac{R-x}{d},B=\frac{R+x}{d}$
　　则 $A<B$
　　代入原式 $${y^2\over d^2}=AB$$
　　显然 $\frac{y^2}{b^2}=(\frac{a}{b})^2$是完全平方数
　　因此 $AB$也是完全平方数
　　又因 $gcd(A,B)=1$，所以 $A$、$B$分别是完全平方数
　　令 $A=a^2,B=b^2$
　　则 $$a^2=\frac{R-x}{d},b^2=\frac{R+x}{d}$$
　　 $$a^2+b^2=\frac{2R}{d}$$
　　由前面易知 $a<b$，所以 $a^2<\frac{R}{d}$
　　因此枚举 $d|2R$，这个可以根号。
　　对于一个 $d|2R$的 $d$，对应两个式子 $$a^2+b^2=\frac{2R}{d}\\a^2+b^2=d$$
　　其中分别有 $a^2<\frac{R}{d}$和 $a^2<\frac{d}{2}$
　　枚举 $a^2$，算出对应的 $b$，如果$b$是完全平方数且 $gcd(a,b)=1$，那就找到了一对点，这对点可以倒着解回去， $x=\sqrt{R-a^2d}$, $y=\sqrt{R^2-x^2}$
　　在 $will7101$大佬的帮助下，用积分求出时间复杂度 $O(n^{\frac{3}{4}})$

代码

//数学 
#include <cstdio>
#include <algorithm>
#include <cmath>
#define ll long long
using namespace std;
ll R;
ll gcd(ll a, ll b){return !b?a:gcd(b,a%b);}
bool check(ll x)
{
    ll t=sqrt(x);
    return t*t==x;
}
int main()
{
    ll d, a, b, ans=0;
    scanf("%lld",&R);
    for(d=1;d*d<=2*R;d++)
    {
        if(2*R%d==0)
        {
            for(a=1;a*a<d/2;a++)
            {
                b=sqrt(d-a*a);
                if(a*a+b*b==d and gcd(a,b)==1)ans++;
            }
            for(a=1;a*a<R/d;a++)
            {
                b=sqrt(2*R/d-a*a);
                if(a*a+b*b==2*R/d and gcd(a,b)==1)ans++;
            }
        }
    }
    printf("%lld\n",ans*4+4);
    return 0;
}