【51nod1678】lyk与gcd(莫比乌斯反演+枚举因数)

点此看题面

大致题意: 一个长度为\(n\)的数组,实现两个操作:单点修改,给定\(i\)\(\sum_{j=1}^na_j[gcd(i,j)=1]\)

莫比乌斯反演

考虑推一推询问操作的式子:

\[\sum_{j=1}^na_j[gcd(i,j)=1]\]

按照莫比乌斯反演的一般套路,我们知道\(\sum_{p|x}\mu(p)=[x=1]\),因此我们枚举一个\(p\)

\[\sum_{j=1}^na_j\sum_{p|i,p|j}\mu(p)\]

调整枚举顺序,得到:

\[\sum_{p|i}\mu(p)\sum_{j=1}^na_j[p|j]\]

考虑到一个数的约数个数很少,所以我们可以直接枚举\(p\),然后只要维护满足\(p|j\)\(a_j\)之和,就可以求出答案。

则我们可以发现,同样因为一个数的约数个数很少,单点修改时,我们可以枚举所修改位置的编号的约数并修改每个约数的答案,这样就能实现维护了。

代码

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 100000
#define IT vector<int>::iterator
#define pb push_back
using namespace std;
int n,a[N+5],s[N+5];vector<int> v[N+5];
class FastIO
{
    private:
        #define FS 100000
        #define tc() (A==B&&(B=(A=FI)+fread(FI,1,FS,stdin),A==B)?EOF:*A++)
        #define pc(c) (C==E&&(clear(),0),*C++=c)
        #define tn (x<<3)+(x<<1)
        #define D isdigit(c=tc())
        int T;char c,*A,*B,*C,*E,FI[FS],FO[FS],S[FS];
    public:
        I FastIO() {A=B=FI,C=FO,E=FO+FS;}
        Tp I void read(Ty& x) {x=0;W(!D);W(x=tn+(c&15),D);}
        Tp I void write(Ty x) {W(S[++T]=x%10+48,x/=10);W(T) pc(S[T--]);}
        Tp I void writeln(Con Ty& x) {write(x),pc('\n');}
        I void clear() {fwrite(FO,1,C-FO,stdout),C=FO;} 
}F;
class LinearSieve//线性筛预处理莫比乌斯函数
{
    private:
        int Pt,P[N+5],mu[N+5];
    public:
        I int operator [] (CI x) Con {return mu[x];}
        I LinearSieve()
        {
            mu[1]=1;for(RI i=2,j;i<=N;++i)
                for(!P[i]&&(mu[P[++Pt]=i]=-1),j=1;j<=Pt&&1LL*i*P[j]<=N;++j)
                    if(P[i*P[j]]=1,i%P[j]) mu[i*P[j]]=-mu[i];else break;
        }
}L;
int main()
{
    RI Qt,i,j,op,x,y,t;IT it;for(F.read(n),F.read(Qt),i=1;i<=n;++i)
    {
        for(F.read(a[i]),j=1;1LL*j*j<=i;++j) !(i%j)&&(v[i].pb(j),i^(j*j)&&(v[i].pb(i/j),0));//预处理约数
        for(it=v[i].begin();it!=v[i].end();++it) s[*it]+=a[i];//预处理答案
    }
    W(Qt--) switch(F.read(op),F.read(x),op)
    {
        case 1:for(F.read(y),it=v[x].begin();it!=v[x].end();++it) s[*it]+=y-a[x];a[x]=y;break;//单点修改,枚举约数进行修改
        case 2:for(t=0,it=v[x].begin();it!=v[x].end();++it) t+=L[*it]*s[*it];F.writeln(t);break;//询问,枚举约数统计答案
    }return F.clear(),0;
}

猜你喜欢

转载自www.cnblogs.com/chenxiaoran666/p/51nod1678.html