【51Nod1678】lyk与gcd

这天，lyk又和gcd杠上了。
它拥有一个n个数的数列，它想实现两种操作。

1：将 ai 改为b。
2：给定一个数i，求所有 gcd(i,j)=1 时的 aj 的总和。
Input
第一行两个数n，Q(1<=n,Q<=100000)。
接下来一行n个数表示ai(1<=ai<=10^4)。
接下来Q行，每行先读入一个数A(1<=A<=2)。
若A=1，表示第一种操作，紧接着两个数i和b。(1<=i<=n,1<=b<=10^4)。
若B=2，表示第二种操作，紧接着一个数i。(1<=i<=n)。
Output
对于每个询问输出一行表示答案。
Input示例
5 3
1 2 3 4 5
2 4
1 3 1
2 4
Output示例
9
7

题解
定义sum[i]=a[i]+a[i*2]+a[i*3]+……+a[i*k]
对于询问i，我们求出它的因数，利用容斥原理，加加减减计算答案

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<set>
#include<ctime>
#include<vector>
#include<cmath>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#define mod 1000000007
#define ll long long 
#define N 25
#define inf 0x7fffffff
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int miu[100005],sum[100005],a[100005];
int p[100005],tot,n,Q;
bool flag[100005];
void pre()
{
    miu[1]=1;
    for (int i=2;i<=n;i++)
    {
        if (!flag[i]) p[++tot]=i,miu[i]=-1;
        for (int j=1;j<=tot&&p[j]*i<=n;j++)
        {
            flag[i*p[j]]=1;
            if (i%p[j]==0) break;
            miu[i*p[j]]=-miu[i];
        }
    }
}
int main()
{
    n=read();Q=read();pre();
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=n;i++)
        for (int j=i;j<=n;j+=i) sum[i]+=a[j];
    while (Q--)
    {
        int opt=read();
        if (opt==1)
        {
            int p=read(),x=read();
            for (int i=1;i*i<=p;i++)
                if (p%i==0)
                {
                    sum[i]+=x-a[p];
                    if (i*i!=p) sum[p/i]+=x-a[p];
                }
            a[p]=x;
        }
        else
        {
            int p=read(),q=p;
            ll ans=0;
            for (int i=1;i*i<=p;i++)
                if (p%i==0)
                {
                    ans+=miu[i]*sum[i];
                    if (i*i!=p) ans+=miu[p/i]*sum[p/i];
                }
            printf("%lld\n",ans);
        }
    }
    return 0;
}