BZOJ 1041 HAOI 2008 圆上的整点 数学推导

如果某整数$x$代入圆方程得到的$y$是整数就计入。
坐标轴上必有4个点，而且四象限对称，所以规定$x,y\geq 1$
$\because x^2+y^2=r^2$
$\therefore y^2=r^2-x^2$
$\therefore y^2=(r-x)(r+x)$
$\therefore y=\sqrt{(r-x)(r+x)}$
$令d=\gcd(r-x,r+x),A=\frac{r-x}{d},B=\frac{r+x}{d}$
$\therefore gcd(A,B)=1$
$\therefore y=d\sqrt{AB}$
$\therefore AB为完全平方数$
$\because A\neq B$
$\therefore A,B是完全平方数$
$令a=\sqrt{A},b=\sqrt{B},a<b$
$\therefore a^2=\frac{r-x}{d},b^2=\frac{r+x}{d}$
$\therefore a^2+b^2=\frac{2r}{d}$
$\because a,b是整数$
$\therefore d|2r$
于是$d\in[1,\sqrt{2r}],a\in[1,\sqrt{\frac{d}{2}}]$
枚举并验证是否存在$b$即可。

#include <cstdio>
#include <cmath>
#define FOR(i,j,k) for(i=j;i<=k;++i)
typedef long long ll;
ll gcd(ll a, ll b) { return !b ? a : gcd(b, a % b); }
void chk(ll x, ll &ans) {
    ll sx = sqrt(x / 2), a2, b, b2, a;
    FOR(a,1,sx) {
        a2 = a * a; b2 = x - a2; b = sqrt(b2);
        if (b * b == b2 && gcd(a2, b2) == 1 && a2 != b2) ++ans;
    }
}
int main() {
    ll s2r, d, a, b, ans = 0, r;
    scanf("%I64d", &r);
    s2r = sqrt(2 * r);
    FOR(d,1,s2r) if ((2 * r) % d == 0) chk(d, ans), chk(2 * r / d, ans);
    printf("%I64d", ans * 4 + 4); 
    return 0;
}

1041: [HAOI2008]圆上的整点

Description

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

Input

$r$

Output

整点个数

Sample Input

4

Sample Output

4

HINT

$n\leq 2\times 10^9$