【JZOJ3623】【SDOI2014】数表(table) 树状数组+离线+莫比乌斯反演

题面

100

\[ Ans=\sum_{i=1}^n\sum_{j=1}^mg(gcd(i,j)) \]
其中，
\[ g(d)=\sum_{i|d}i \]
我们注意到\(gcd(i,j)\)最多有\(O(n)\)种取值，所以我们枚举\(d=gcd(i,j)\)；
就有，
\[ Ans=\sum_{d=1}^ng(d)*f(d) \]
其中，\(f(d)\)表示，有多少对\((i,j)\)的最大公约数为\(d\)，可以使用莫比乌斯反演求出。
那么，
\[ Ans=\sum_{d=1}^ng(d)*\sum_{d|i}\lfloor\frac{n}{i}\rfloor\lfloor\frac{m}{i}\rfloor\mu(\frac{i}{d}) \]
更换主体，
\[ Ans=\sum_{i=1}^n\lfloor\frac{n}{i}\rfloor\lfloor\frac{m}{i}\rfloor\sum_{d|i}g(d)*\mu(\frac{i}{d}) \]
令\(h(d)=\sum_{d|i}g(d)*\mu(\frac{i}{d})\)，
对于单个询问而言，如果我们知道\(h\)的前缀和，那么就可以\(O(\sqrt n)\)求出答案。
如果\(a\)一定时，我们可以预处理出\(h\)。
但实际上\(a\)不定。
突破口则是本题允许离线；
我们可以对\(a\)偏序，顺便利用树状数组\(c\)维护\(h\)，具体是：
对于一个\(g(d)<=a\)的，我们给所有\(c[i*d]\)加上一个\(g(d)*\mu(i)\)，然后\(c\)就是当前的\(h\)，
其总的复杂度是\(O(\sum_{i=1}^n\frac{n}{i}*logn)\approx O(nlog^2n)\)。
加上询问，那么最终的复杂度是\(O(Q\sqrt nlogn+nlog^2n)\)

Code

#include<bits/stdc++.h>
#define ll long long
#define fo(i,x,y) for(int i=x;i<=y;i++)
#define fd(i,x,y) for(int i=x;i>=y;i--)
using namespace std;
const char* fin="ex3623.in";
const char* fout="ex3623.out";
const int inf=0x7fffffff;
const int maxn=100007;
const ll mo=1ll<<31;
int t,mu[maxn],wu[maxn],p[maxn],ans[maxn];
int c[maxn],g[maxn];
bool bz[maxn];
void modify(int v,int u){for(;v<maxn;v+=v&-v)c[v]=(0ll+u+c[v])%mo;}
int getsum(int v){
    ll k=0;
    for(;v;v-=v&-v) k+=c[v];
    return k%mo;
}
int getsum(int l,int r){return (getsum(r)-getsum(l-1)+mo)%mo;}
struct query{
    int n,m,a,id;
}q[maxn];
bool cmp(query a,query b){return a.a<b.a;}
bool cmp1(int a,int b){return wu[a]<wu[b];}
int main(){
    freopen(fin,"r",stdin);
    freopen(fout,"w",stdout);
    scanf("%d",&t);
    fo(i,1,t) scanf("%d%d%d",&q[i].n,&q[i].m,&q[i].a),q[i].id=i;
    mu[1]=1;
    wu[1]=1;
    fo(i,2,maxn-1){
        if (!bz[i]){
            p[++p[0]]=i;
            wu[i]=i+1;
            mu[i]=-1;
        }
        fo(j,1,p[0]){
            int k=p[j]*i;
            if (k>=maxn) break;
            bz[k]=true;
            if (i%p[j]==0){
                mu[k]=0;
                wu[k]=wu[i]+p[j]*(wu[i]-wu[i/p[j]]);
                break;
            }else{
                mu[k]=-mu[i];
                wu[k]=p[j]*wu[i]+wu[i];
            }
        }
    }
    fo(i,1,maxn-1) g[i]=i;
    sort(q+1,q+t+1,cmp);
    sort(g+1,g+maxn,cmp1);
    int l=1;
    fo(i,1,t){
        while (l<maxn && wu[g[l]]<=q[i].a){
            for(int k=g[l];k<maxn;k+=g[l]) modify(k,wu[g[l]]*mu[k/g[l]]);
            l++;
        }
        int n=q[i].n,m=q[i].m;
        if (n>m) swap(n,m);
        for(int l=1,r;l<=n;l=r+1){
            r=min(n/(n/l),m/(m/l));
            ans[q[i].id]=(ans[q[i].id]+1ll*getsum(l,r)*(n/l)*(m/l))%mo;
        }
    }
    fo(i,1,t) printf("%d\n",ans[i]);
    return 0;
}

转载于:https://www.cnblogs.com/hiweibolu/p/6913677.html