【HDU4947】GCD Array (莫比乌斯反演+树状数组)

BUPT2017 wintertraining(15) #5H

HDU- 4947

题意

有一个长度为l的数组,现在有m个操作,第1种为1 n d v,给下标x 满足gcd(x,n)=d的(a_x)增加v。第2种为2 x,查询(sum_{i=1}^x a_i)

数据范围:(1le n,d,vle2cdot 10^5,1le xle l)

题解

(f_i)满足(a_i=sum_{d|i} f_d),用树状数组存储(f_i)的前缀和。

[a_x+=vcdot[gcd(x,n)=d]=vcdot[gcd(x/d,n/d)=1] ]

根据莫比乌斯函数(关于莫比乌斯反演可以看这篇论文:贾志鹏线性筛法与积性函数)的性质,我们知道(sum_{d|n}mu(d)=[n=1]),(这个d和上面的d不是同一个,下面换为p表示) 因此

[a_x+=vcdotsum_{p|gcd(x/d,n/d)}mu(p)=vcdot sum_{p|frac x d,p|frac n d}mu(p) ]

于是对于给定的n和d,(frac n d)的因子p的d倍就是符合条件的下标x的一个因子。

莫比乌斯反演可得:

(f(n)=sum_{d|n}mu(frac n d)a(n))

因此(f_{pd}=sum mu(p)a(pd)),于是对于1操作,我们只要给(f_{pd})加上(vcdot mu (p))即可。

2操作,是对(a_i)求和:

[sum_{i=1}^x a_i=sum_{i=1}^x sum_{d|x}f_d ]

对于固定的d来说,1~x内(f_d)要加(lfloor frac x d floor)次。再分块加速一下,也就是对于(lfloorfrac x d floor)相同的d,把(f_d)区间和求出来再乘上(lfloorfrac x d floor),设这个区间是[d1,d2],那么d2=x/(x/d1) (整除),为什么呢?因为d2是满足(frac x d ge lfloor frac x {d1} floor=k)的最大的整数d,那么(xge d2cdot k),所以(frac x k ge d2),也就是d2=(lfloorfrac x k floor)

这题的时间复杂度:

预处理出1~N的所有因子,(O(nsqrt n))

计算莫比乌斯函数,(O(n))

1操作,因子有(sqrt n)个,增加是(O(log n)),总的是(O(msqrt n log n))

2操作,查询(O(log n)),分块(O(sqrt n)),也是(O(msqrt n log n))

总的就是(O(msqrt n log n))

官方题解:

img

代码

#include<cstdio>
#include<cstring>
#include<vector>
#define ll long long
#define N 200005
using namespace std;

int miu[N],prime[N],cnt;
ll sum[N],last,lasttemp,temp;
vector<int>fac[N];
bool check[N];
ll ans;
ll getsum(int x){
    ll ans=0;
    for(;x;x-=x&-x)ans+=sum[x];
    return ans;
}
void add(int x,int v){
    for(;x<N;x+=x&-x)sum[x]+=v;
}
void Mobius(){
    miu[1]=1;
    for(int i=2;i<N;i++){
        if(!check[i]){
            prime[cnt++]=i;
            miu[i]=-1;
        }
        for(int j=0;j<cnt;j++){
            if(i*prime[j]>N)break;
            check[i*prime[j]]=1;
            if(i%prime[j])miu[i*prime[j]]=-miu[i];
            else break;
        }
    }
}
int main(){
    int l,m,cas=0;
    Mobius();
    for(int i=1;i<N;i++)
        for(int j=i;j<N;j+=i)
            fac[j].push_back(i);
    while(scanf("%d%d",&l,&m),l,m){
        printf("Case #%d:
",++cas);
        memset(sum,0,sizeof sum);
        while(m--){
            int n,d,v,x;
            scanf("%d",&n);
            if(n==1){
                scanf("%d%d%d",&n,&d,&v);
                if(n%d)continue;
                n/=d;
                for(int i=0;i<fac[n].size();i++){
                    x=fac[n][i];
                    add(x*d,miu[x]*v);
                }
            }else{
                scanf("%d",&n);
                ans=temp=0;
                for(int i=1;i<=n;i=last+1){
                    last=n/(n/i);
                    lasttemp=temp;
                    temp=getsum(last);
                    ans+=n/i*(temp-lasttemp);
                }
                printf("%lld
",ans);
            }
        }
    }
}

相似题,待做: SPOJ PGCD - Primes in GCD Table (好题! 莫比乌斯反演+分块求和优化)

待看的文章:读贾志鹏线性筛有感 (莫比乌斯函数的应用)

原文地址:https://www.cnblogs.com/flipped/p/HDU4947.html