bzoj3529 [ SDOI2014 ] -- 莫比乌斯反演+分块

题目大意：

求

$$\sum_{i=1}^n\sum_{j=1}^m\sigma(\gcd(i,j))[\sigma(\gcd(i,j))\le a]$$

其中$\sigma(x)$表示$x$的约数和。
先假设没有$a$的限制，且$n\le m$。
那么就是求

$$\sum_{i=1}^n\sum_{j=1}^m\sigma(\gcd(i,j))$$

$$\sum_{i=1}^n\sum_{j=1}^m\sum_{d=1}^n\sigma(d)[d=\gcd(i,j)]$$

$$\sum_{d=1}^n\sigma(d)\sum_{i=1}^{\lfloor{n\over d}\rfloor}\sum_{j=1}^{\lfloor{m\over d}\rfloor}[\gcd(i,j)=1]$$

$$\sum_{d=1}^n\sigma(d)\sum_{i=1}^{\lfloor{n\over d}\rfloor}\sum_{j=1}^{\lfloor{m\over d}\rfloor}\sum_{t|i,t|j}\mu{(t)}$$

$$\sum_{d=1}^n\sigma(d)\sum_{t=1}^{\lfloor{n\over d}\rfloor}\mu{(t)} \lfloor{n\over dt}\rfloor \lfloor{m\over dt}\rfloor$$

令$T=dt$

$$\sum_{T=1}^n\lfloor{n\over T}\rfloor \lfloor{m\over T}\rfloor\sum_{d|T}\sigma(d)\mu{({T\over d})}$$

令$f(x)=\sum_{d|x}\sigma(d)\mu{({x\over d})}$

$$\sum_{T=1}^n\lfloor{n\over T}\rfloor \lfloor{m\over T}\rfloor f(T)$$

那么$O(n\log n)$求出$f$，就能$O(q\sqrt n)$求出答案了。
然后考虑有$a$的限制，那么

$$f(x)=\sum_{d|x}\sigma(d)\mu{({x\over d})}[\sigma{(d)}\le a]$$

离线，将询问按照$a$从小到大排序，每次用$a_{i-1}+1\le\sigma{(d)}\le a_{i}$的$\sigma{(d)}$更新$f$，用树状数组维护前缀和就可以了。
时间复杂度$O(n\log^2n+q\sqrt n\log n)$

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long
#define M 2147483648
#define N 100010
struct Qu{
    int x,y,z,f;
}q[20010];
struct Node{
    int w,f;
}a[N];
ll c[N],f[N],Ans[N];
int Q,i,j,k,n,m,p[N],Top,mu[N],Last;
bool b[N];
inline bool Cmp2(Node a,Node b){
    return a.w<b.w;
}
inline bool Cmp(Qu a,Qu b){
    return a.z<b.z;
}
inline void Init(){
    mu[1]=1;
    for(i=2;i<N;i++){
        if(!b[i]){
            p[++Top]=i;
            mu[i]=-1;
        }
        for(j=1;j<=Top&&p[j]*i<N;j++){
            b[p[j]*i]=1;
            if(!(i%p[j])){mu[p[j]*i]=0;break;}
            mu[p[j]*i]=-mu[i];
        }
    }
    for(i=1;i<N;i++)
    for(j=1;j*i<N;j++)
    f[i*j]+=i;
    for(i=1;i<N;i++)a[i].w=f[i],a[i].f=i;
    sort(a+1,a+N,Cmp2);
}
inline void Update(int x,ll y){
    for(;x<N;x+=x&-x)c[x]=(c[x]+y)%M;
}
inline ll Query(int x){
    ll Ans=0;
    for(;x;x-=x&-x)Ans+=c[x];
    return Ans;
}
inline void Update1(int x){
    for(int i=1;i*x<N;i++)Update(i*x,f[x]*mu[i]);
}
inline int Min(int x,int y){
    return x<y?x:y;
}
inline ll GetAns(int n,int m){
    ll Ans=0;
    for(int i=1;i<=n;){
        int j=Min(n/(n/i),m/(m/i));
        Ans=(Ans+(Query(j)-Query(i-1))*(n/i)%M*(m/i)%M)%M;
        i=j+1;
    }
    return Ans;
}
int main(){
    scanf("%d",&Q);
    for(i=1;i<=Q;i++){
        scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].z);
        if(q[i].x>q[i].y)swap(q[i].x,q[i].y);
        q[i].f=i;
    }
    Init();
    sort(q+1,q+Q+1,Cmp);
    for(i=1;i<=Q;i++){
        while(Last<N-1&&a[Last+1].w<=q[i].z)Update1(a[++Last].f);
        Ans[q[i].f]=GetAns(q[i].x,q[i].y);
    }
    for(i=1;i<=Q;i++)printf("%lld\n",(Ans[i]+M)%M);
    return 0;
}