BZOJ 2154: Crash的数字表格（莫比乌斯反演）

题目描述

传送门

题目大意：有一个n*m的表格，第i行第j列的数是lcm(i,j)，求表格内所有数的和对20101009取模的结果（n，m<=10^7）。

---

题解

简单的数论题。老套路，先假设$n<=m$。题目要求

$$\sum_{i=1}^n\sum_{j=1}^m[i,j]$$

画一下柿子

$$ans=\sum_{i=1}^n\sum_{j=1}^m{ij\over(i,j)}$$
$$=\sum_{d=1}^n\sum_{i=1}^n\sum_{j=1}^m[(i,j)=d]{ij\over d}$$
$$=\sum_{d=1}^nd\sum_{i=1}^{\lfloor{n\over d}\rfloor}\sum_{j=1}^{\lfloor{m\over d}\rfloor}[(i,j)=1]ij$$
令 $F(x,y)=\sum_{i=1}^x\sum_{j=1}^y[(i,j=1)]ij$，则
$$ans=\sum_{d=1}^ndF(\lfloor{n\over d}\rfloor,\lfloor{m\over d}\rfloor)$$
假如我们已经预先求出了 $F$，明显这里分块优化一下，维护$d$的前缀和，求出答案的时间是 $O(\sqrt n)$的。

但发现好像不能预处理$F$（$F$是二元函数），所以只能在线去算。

我们将$F$反演一下

$$F(x,y)=\sum_{i=1}^x\sum_{j=1}^yij\sum_{t|i,t|j}\mu(t)$$

$$=\sum_{t=1}^xt^2\mu(t)\sum_{i=1}^{\lfloor{x\over t}\rfloor}\sum_{j=1}^{\lfloor{y\over t}\rfloor}ij$$

我们搞一个新的函数$Sum(x,y)=\sum_{i=1}^x\sum_{j=1}^yij={x(x+1)\over2}{y(y+1)\over2}$

于是

$$F(x,y)=\sum_{t=1}^xt^2\mu(t)Sum({\lfloor{x\over t}\rfloor},{\lfloor{y\over t}\rfloor})$$

我们预处理出$i^2\mu(i)$的前缀和，再搞一个分块就可以求出$F$了，于是总时间就是$O(\sqrt n)*O(\sqrt n)=O(n)$。

有时候设出函数，分步化简求解比一步算到尽头要更清晰和简单。

对于单组询问，这样做就可以了。如果是多组询问呢？那就需要更秀的操作，这里就不说了，以后有空再去做加强版吧。

---

代码

#include <bits/stdc++.h>
#define maxn 10000005
#define MOD 20101009
#define temp (i * prime[j])

using namespace std;

int n, m, cnt;
int prime[maxn], miu[maxn], sd[maxn];
bool Vis[maxn];

void Da(){
    miu[1] = 1;
    for(int i = 2; i <= n; i++){
        if(!Vis[i]){
            prime[++cnt] = i;
            miu[i] = -1;    
        }
        for(int j = 1; j <= cnt && temp <= n; j++){
            Vis[temp] = true;
            if(i % prime[j] == 0){
                miu[temp] = 0;
                break;
            }
            else  miu[temp] = -miu[i];
        }
    }
    for(int i = 1; i <= n; i++)  miu[i] = 1LL * miu[i] * i * i % MOD;
    for(int i = 2; i <= n; i++)  miu[i] = (miu[i] + miu[i-1]) % MOD;
    for(int i = 1; i <= n; i++)  sd[i] = i;
    for(int i = 2; i <= n; i++)  sd[i] = (sd[i] + sd[i-1]) % MOD;
}

int Sum(int x, int y){
    int temp1 = (1LL * x * (x + 1) >> 1) % MOD;
    int temp2 = (1LL * y * (y + 1) >> 1) % MOD;
    return 1LL * temp1 * temp2 % MOD;
}

int F(int x, int y){
    if(x > y)  swap(x, y);
    int res = 0;
    int last;

    for(int i = 1; i <= x; i = last+1){
        last = min(x/(x/i), y/(y/i));
        res = (res + 1LL * (miu[last] - miu[i-1] + MOD) % MOD * Sum(x/i, y/i) % MOD) % MOD;
    }
    return res;
}

int Solve(){
    int ans = 0;
    int last;

    for(int i = 1; i <= n; i = last+1){
        last = min(n/(n/i), m/(m/i));
        ans = (ans + 1LL * (sd[last] - sd[i-1] + MOD) % MOD * F(n/i, m/i) % MOD) % MOD;
    }
    return ans;
}   

int main(){

    scanf("%d%d", &n, &m);

    if(n > m)  swap(n, m);

    Da();

    printf("%d\n", Solve());

    return 0;
}//注意乘法可能爆int

---