51nod 1678 lyk与gcd

记录每个i的素因子，以及每个i的倍数j的a[j]之和。容斥一下。由于只考虑是否互质，因此容斥时考虑素因子即可。

#include<bits/stdc++.h>
using namespace std;

const int MAXN=100100;
vector<int> fac[MAXN];
int sum[MAXN],npri[MAXN],a[MAXN];

int cal1(int x)
{
	int ret=0;
	while(x)
	{
		if(x&1)
			ret++;
		x>>=1;
	}
	return ret;
}

int main()
{
	int n,q,i,j,tmp,ans,op,x,b,dis,limit;
	while(~scanf("%d%d",&n,&q))
	{
		memset(sum,0,sizeof(sum));
		memset(npri,0,sizeof(npri));
		for(i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			sum[1]+=a[i]; 
		}
		for(i=2;i<=n;i++)
		{
			for(j=i;j<=n;j+=i)
			{
				if(!npri[i]&&(!fac[j].size()||fac[j][fac[j].size()-1]!=i)) 
					fac[j].push_back(i);
				sum[i]+=a[j];
				if(j>i)
					npri[j]|=1;
			}
		}
		while(q--)
		{
			scanf("%d",&op);
			if(op==1)
			{
				scanf("%d%d",&x,&b);
				dis=b-a[x];
				a[x]=b;
				limit=sqrt(x);
				for(i=1;i<=limit;i++)
				{
					if(x%i==0)
					{
						sum[i]+=dis;
						sum[x/i]+=dis;
					}
				}
				if(limit*limit==x)
					sum[limit]-=dis;
			}
			else
			{
				scanf("%d",&x);
				ans=sum[1];
				limit=1<<(fac[x].size());
				for(i=1;i<limit;i++)
				{
					tmp=1;
					for(j=0;j<fac[x].size();j++)
					{
						if((1<<j)&i)
							tmp*=fac[x][j];
					}
					if(cal1(i)&1)
						ans-=sum[tmp];
					else
						ans+=sum[tmp];
				}
				printf("%d\n",ans);
			}
		}
	}
}