[NOI 2010]能量采集

Description

给你一个 \(n\times m\) 的坐标轴。对于坐标轴的每一个正整数整点 \((x,y)\) 其对答案产生的贡献为 \(2k+1\) ，其中 \(k\) 表示这个点与坐标原点连线，线段穿过了除端点外的 \(k\) 个点。求所有点的贡献和。

\(1\leq n,m \leq 100000\)

Solution

容易发现 \(k=\gcd(x,y)-1\) ，故原式等于求 \[\begin{aligned}&\sum_{i=1}^n\sum_{j=1}^m(2(\gcd(i,j)-1)+1)\\=&2\sum_{i=1}^n\sum_{j=1}^m\gcd(i,j)-nm\end{aligned}\]

记 \(f(n,m)=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\gcd(i,j)\) ，考虑如何求 \(f(n,m)\) \[\begin{aligned}\Rightarrow&\sum_{d=1}^{min\{n,m\}}d\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}[\gcd(i,j)=1]\\=&\sum_{d=1}^{min\{n,m\}}d\sum_{i=1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j=1}^{\left\lfloor\frac{m}{d}\right\rfloor}\sum_{k\mid \gcd(i,j)}\mu(k)\\=&\sum_{d=1}^{min\{n,m\}}d\sum_{k=1}^{min\left\{\left\lfloor\frac{n}{d}\right\rfloor,\left\lfloor\frac{m}{d}\right\rfloor\right\}}\mu(k)\left\lfloor\frac{n}{kd}\right\rfloor\left\lfloor\frac{m}{kd}\right\rfloor\end{aligned}\]

令 \(T=kd\) ，枚举 \(T\) \[\begin{aligned}\sum_{T=1}^{min\{n,m\}}\left\lfloor\frac{n}{T}\right\rfloor\left\lfloor\frac{m}{T}\right\rfloor\sum_{d\mid T}d\cdot\mu\left(\frac{T}{d}\right)\end{aligned}\]

记后面那个狄利克雷卷积形式的式子为 \(F(T)\) ，显然这个是可以枚举因子在近似于 \(O(n\ln n)\) 的时限内预处理出来的。然后数论分块的复杂度为 \(O(\sqrt n)\) ，对于 \(t\) 组询问...哦...没有 \(t\) 组询问...~~那我最后一步还搞个屁啊...~~

Code

//It is made by Awson on 2018.2.22

#include <bits/stdc++.h>

#define LL long long

#define dob complex<double>

#define Abs(a) ((a) < 0 ? (-(a)) : (a))

#define Max(a, b) ((a) > (b) ? (a) : (b))

#define Min(a, b) ((a) < (b) ? (a) : (b))

#define Swap(a, b) ((a) ^= (b), (b) ^= (a), (a) ^= (b))

#define writeln(x) (write(x), putchar('\n'))

#define lowbit(x) ((x)&(-(x)))

using namespace std;

const int N = 100000;

void read(int &x) {

    char ch; bool flag = 0;

    for (ch = getchar(); !isdigit(ch) && ((flag |= (ch == '-')) || 1); ch = getchar());

    for (x = 0; isdigit(ch); x = (x<<1)+(x<<3)+ch-48, ch = getchar());

    x *= 1-2*flag;

}

void print(LL x) {if (x > 9) print(x/10); putchar(x%10+48); }

void write(LL x) {if (x < 0) putchar('-'); print(Abs(x)); }

int n, m, mu[N+5];

LL F[N+5];

void get_F() {

    int isprime[N+5], prime[N+5], tot = 0;

    memset(isprime, 1, sizeof(isprime)); isprime[1] = 0, mu[1] = 1;

    for (int i = 2; i <= N; i++) {

    if (isprime[i]) mu[i] = -1, prime[++tot] = i;

    for (int j = 1; j <= tot && i*prime[j] <= N; j++)

        if (i%prime[j] != 0) isprime[i*prime[j]] = 0, mu[i*prime[j]] = -mu[i];

        else {isprime[i*prime[j]] = 0, mu[i*prime[j]] = 0; break; }

    }

    for (int i = 1; i <= N; i++) for (int j = 1; j*i <= N; j++) F[i*j] += i*mu[j];

    for (int i = 1; i <= N; i++) F[i] += F[i-1];

}

LL cal(int n, int m) {

    if (n > m) Swap(n, m); LL ans = 0;

    for (int i = 1, last; i <= n; i = last+1) {

    last = Min(n/(n/i), m/(m/i));

    ans += 1ll*(n/i)*(m/i)*(F[last]-F[i-1]);

    }

    return ans;

}

void work() {

    read(n), read(m); get_F(); writeln(2ll*cal(n, m)-1ll*n*m);

}

int main() {

    work(); return 0;

}