HDU4947GCD Array（莫比乌斯反演+树状数组）

题面

传送门

题解

orz ljz

相当于每一个数要加上

\[v\times [\gcd(i,n)=d]=v\times [\gcd(i/d,n/d)=1]=v\times \sum_{p|{i\over d},p|{n\over d}}\mu(p)\]

那么我们可以维护一个\(f_i\)，每次令\(p|{n\over d}\)的\(f_{p\times d}\)加上\(v\)，这样\(a_i=\sum\limits_{p|i}f_p\)，同时修改的数字可以大大减少

然后现在怎么求和呢？我们有\(\sum\limits_{i=1}^na_i=\sum\limits_{i=1}^n{\lfloor{n\over i}\rfloor}f_i\)，小于\(\sqrt{n\log n}\)的部分暴力，大的部分整除分块并用树状数组维护即可

//minamoto
#include<bits/stdc++.h>
#define R register
#define pb push_back
#define inline __inline__ __attribute__((always_inline))
#define fp(i,a,b) for(R int i=(a),I=(b)+1;i<I;++i)
#define fd(i,a,b) for(R int i=(a),I=(b)-1;i>I;--i)
#define go(u) for(int i=head[u],v=e[i].v;i;i=e[i].nx,v=e[i].v)
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
using namespace std;
char buf[1<<21],*p1=buf,*p2=buf;
inline char getc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;}
int read(){
    R int res,f=1;R char ch;
    while((ch=getc())>'9'||ch<'0')(ch=='-')&&(f=-1);
    for(res=ch-'0';(ch=getc())>='0'&&ch<='9';res=res*10+ch-'0');
    return res*f;
}
const int N=2e5+5;
typedef long long ll;
vector<int>vec[N];int mu[N],p[N],vis[N],n,q,m,T;ll c[N],f[N];
inline void add(R int x,R int y){for(;x<=n;x+=x&-x)c[x]+=y;}
inline ll query(R int x){R ll res=0;for(;x;x-=x&-x)res+=c[x];return res;}
void init(int n=N-5){
    mu[1]=1;
    fp(i,2,n){
        if(!vis[i])p[++m]=i,mu[i]=-1;
        for(R int j=1;j<=m&&1ll*i*p[j]<=n;++j){
            vis[i*p[j]]=1;
            if(i%p[j]==0)break;
            mu[i*p[j]]=-mu[i];
        }
    }
    fp(i,1,n)if(mu[i])for(R int j=i;j<=n;j+=i)vec[j].pb(i);
}
int main(){
//  freopen("testdata.in","r",stdin);
    int lim=400;init();
    for(T=0;n=read(),m=read(),n||m;){
        fp(i,1,n)c[i]=f[i]=0;
        printf("Case #%d:\n",++T);
        for(int op,x,d,v,tmp;m;--m){
            op=read(),x=read();
            if(op&1){
                d=read(),v=read();if(x%d)continue;
                fp(i,0,vec[x/d].size()-1){
                    tmp=vec[x/d][i]*d;
                    if(tmp<=n)f[tmp]+=mu[tmp/d]*v,add(tmp,mu[tmp/d]*v);
                }
            }else{
                ll las=0,now=0,res=0;
                fp(i,1,min(lim,x))res+=1ll*x/i*f[i],las+=f[i];
                for(R int l=lim+1,r;l<=x;l=r+1)r=x/(x/l),now=query(r),res+=x/l*(now-las),las=now;
                printf("%lld\n",res);
            }
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/bztMinamoto/p/10901405.html