bzoj 1041: [HAOI2008]圆上的整点 （数学）

题目描述

传送门

题目大意：求一个给定的圆(x^2+y^2=r^2)，在圆周上有多少个点的坐标是整数。

题解

$x^2+y^2=r^2$
$x^2=r^2-y^2$
$x=\sqrt{ (r+y)(r-y)}$
设$d=gcd(r+y,r-y),A={r+y\over d},B={r-y\over d}$
式子变形为$x^2=d^2*A*B$
因为x,d都是完全平方数，所以$A*B$也是完全平方数，$A!=B$所以$A,B$分别是完全平方数。
那么设$A=a*a,B=b*b$
$a^2={r+y\over d},b^2={r-y\over d}$
$a^2+b^2={2r\over d}$
那么枚举2r的因数d,再枚举a,判断是否有满足条件的(a,b)，更新答案即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define LL long long 
using namespace std;
LL r,ans;
LL gcd(LL x,LL y)
{
    if (!x||!y) return 0;
    LL r;
    while (y){
        r=x%y;
        x=y; y=r;
    }
    return x;
}
bool check(LL x,LL y)
{
    if (x!=y&&gcd(x,y)==1) return true;
    return false;
}
void solve(LL x)
{
    LL t=2*r/x;
    for (LL i=1;i<=sqrt(t/2.0);i++) {
        LL a=floor((double)sqrt((t-i*i)*1.0));
        if ((a*a+i*i)!=t) continue;
        if(check(a,i)) ans++;
    }
}
int main()
{
    freopen("a.in","r",stdin);
    scanf("%lld",&r);
    for (LL d=1;d*d<=2*r;d++)
     if ((2*r)%d==0) {
        solve(d);
        if (d*d!=r*2) solve(2*r/d);
     }
    printf("%lld\n",ans*4+4);
}