[BZOJ3817][Sum][类欧几里得算法 数论]

题目大意：

给定$N<=10^9,R<=10^4$，求：

$$\sum_{d=1}^{n} (-1)^{\lfloor \sqrt{d*r*d} \rfloor}$$

思路：

不妨设$x=\sqrt r$，那么

$$\begin{align} -1^{\lfloor dx \rfloor} & =1-2(\lfloor dx \rfloor \% 2) \\ &=1-2(\lfloor dx \rfloor - \lfloor \frac{dx}{2} \rfloor * 2) \\ &= 1+4\lfloor \frac{dx}{2} \rfloor + 2\lfloor dx \rfloor\end {align}$$

所以：

$$\sum_{d=1}^{n } =n + 4 \sum_{d=1}^{n} \lfloor \frac{dx}{2} \rfloor -2\sum_{d=1}^{n} \lfloor dx \rfloor$$

考虑表达式的一般形式：

$$\sum_{i=1}^{n} \lfloor \frac{a}{bx+c}i \rfloor = \sum_{i=1}^{n} \lfloor ki \rfloor$$

当$k>=1$时：

$$\begin{align} &= \sum_{i=1}^{n} (\lfloor \frac{bx+c}{a} \rfloor + \frac{bx+c- \lfloor \frac{bx+c}{a}\rfloor a }{a})i \\ &= \sum_{i=1}^{n} ( \frac{bx+c- \lfloor \frac{bx+c}{a}\rfloor a }{a})i + \lfloor \frac{bx+c}{a} \rfloor * C_{n}^{2}\end{align}$$

当$k<1$时：

$$=\lfloor kn \rfloor n - \sum_{j=1}^{kn} \lfloor \frac{abx-ac}{b^2v-c^2} * j\rfloor$$

• 发现新的表达式和原表达式形式类似，因此可以递归求解，辗转的过程类似于欧几里得法求$gcd$，因此被称为类欧。

• 考虑它的几何意义，相当于是求一个斜边为$y=kx$直角三角形内整点的个数，这个过程相当于把三角形不停翻转。

• 因为$k>=1$和$k<1$必定是轮流出现，所以这两步可以在一步内完成。

• 为什么用cin读入会RE？？？

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll T, n, m;
double t;
inline  ll gcd(ll a, ll b) {
    if (!b) return a;
    return gcd(b, a % b);
}
inline ll calc(ll n, ll a, ll b, ll c) {
    if (!n) return 0;
    ll g = gcd(gcd(a, b), c); a /= g, b /= g, c /= g;
    ll k = (t * b + c) / a, ret = n * (n + 1) / 2 * k;
    c -= k * a; k = (t * b + c) / a * n;
    ret += n * k;
    ret -= calc(k, b * b * m - c * c, a * b, -a * c);
    return ret;
}
int main(void) {
    //freopen("in.txt", "r", stdin);
    scanf("%lld", &T);
    while (T--) {
        scanf("%lld%lld", &n, &m); t = sqrt(m);
        if ((ll)t == t) printf("%lld\n", (ll)((m & 1) ? ((n & 1) ? -1 : 0) : n));
        else printf("%lld\n", n + (calc(n, 2, 1, 0) << 2) - (calc(n, 1, 1, 0) << 1));
    }
    return 0;
}

完。

By g1n0st