UOJ42. 【清华集训2014】Sum

传送门

Sol

$(-1{)}^a=1-2(amod2)=1-2a+4\lfloor \frac{a}{2}\rfloor$
那么原式变成 $n-2{\sum }_{i=1}^n\lfloor d\sqrt{r}\rfloor +4{\sum }_{i=1}^n\lfloor \frac{d\sqrt{r}}{2}\rfloor$
考虑计算这样一个东西
$$\sum _{i=1}^n\lfloor \frac{a\ast \sqrt{r}+b}{c}i\rfloor$$
如果 $\sqrt{r}$ 是一个整数，直接 $\Theta (1)$ 计算
否则
设 $k=\frac{a\ast \sqrt{r}+b}{c}$

如果 $k\ge 1$ 那么可以把 $k$ 的整数部分的值算出来，变成 $0<k<1$

如果 $0<k<1$，即就是计算 $y=kx$ 与 $x=n$ 所围成三角形的整点个数

根据类欧几里得那一套理论，我们用矩形的减去左上角的三角形的

矩形的就是 $n\lfloor nk\rfloor$
左上角的三角形的把它关于 $y=x$ 对称，变成求 $y=\frac{x}{k}$ 与 $x=\lfloor nk\rfloor$ 所围成三角形的整点个数

这样可以递归下去，$n$ 的规模减小得很快，最后计算 $n\le 1$ 的答案即可

为了防止精度误差，可以把 $k$ 表示成 $\frac{a\ast \sqrt{r}+b}{c}$ 将 $a,b,c$ 存下来
每次弄走 $gcd$ 就可以了

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

int n, ans, r, test;
double sq;

inline ll Gcd(ll a, ll b) {
	if (!a || !b) return a | b;
	return !b ? a : Gcd(b, a % b);
}

inline ll Solve(ll a, ll b, ll c, ll len) {
	if (!len) return 0;
	register ll sk, nxt, ret, d;
	d = Gcd(Gcd(a, b), c), a /= d, b /= d, c /= d;
	if (len == 1) return (ll)(1.0 * (sq * a + b) / c);
	register double k = 1.0 * (sq * a + b) / c;
	sk = (ll)k, k -= sk, nxt = (ll)(k * len), b -= c * sk;
	ret = len * nxt + sk * (len + 1) * len / 2;
	return ret - Solve(a * c, -b * c, a * a * r - b * b, nxt);
}

int main() {
	scanf("%d", &test);
	while (test--) {
		scanf("%d%d", &n, &r), sq = sqrt(r), ans = sq;
		if (ans * ans == r) printf("%d\n", (ans & 1) ? ((n & 1) ? -1 : 0) : n);
		else printf("%lld\n", n - 2 * Solve(1, 0, 1, n) + 4 * Solve(1, 0, 2, n));
	}
	return 0;
}