bzoj #3853. GCD Array 题解

题目传送门

题目大意： 给出一个序列 $a$，初始每个位置都为 $0$，有两种操作：1、给出 $n,d,v$，让所有满足 $\gcd (n,x)=d$ 的 $a[x]$ 加 $v$；2、求 $a$ 的某个前缀和。

题解

维护一个辅助数组 $f$，满足 $a_i={\sum }_{d|i}{f}_d$。

考虑每个 $a_x$，它在每次操作时的增加量为 $v[(x,n)=d]$，推一推就是：
$$\begin{array}{rl}& v[(x,n)=d]\\ & =v[(\frac{x}{d},\frac{n}{d})=1]\\ & =\sum _{k|(\frac{x}{d},\frac{n}{d})}v\mu (k)\\ & =\sum _{dk|x,k|\frac{n}{d}}v\mu (k)\end{array}$$

可以发现，枚举出一个 $\frac{n}{d}$ 的因子 $k$，那么 $dk$ 一定是满足条件的 $x$ 的因子，所以可以让 $f_{dk}$ 加上 $v\mu (k)$。

再考虑求解，${\sum }_{i=1}^x{a}_i={\sum }_{i=1}^x{\sum }_{j|i}{f}_j={\sum }_{j=1}^n{f}_j\lfloor \frac{x}{j}\rfloor$，于是套一个整除分块即可，这里要求 $f$ 的前缀和，所以要用一个树状数组来维护，复杂度为 $O(n\sqrt{n}\log n)$。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 200010
#define ll long long

int n,m;ll tr[maxn];
void add(int x,int y){for(;x<=maxn-10;x+=(x&-x))tr[x]+=y;}
ll sum(int x){ll re=0;for(;x>=1;x-=(x&-x))re+=tr[x];return re;}
int mu[maxn],prime[maxn],t=0;
bool v[maxn];
struct edge{int y,next;}e[maxn<<4];
int first[maxn],len=0;
void buildroad(int x,int y){e[++len]=(edge){y,first[x]};first[x]=len;}
void work(){
	mu[1]=1;for(int i=2;i<=maxn-10;i++){
		if(!v[i])prime[++t]=i,mu[i]=-1;
		for(int j=1;j<=t&&i*prime[j]<=maxn-10;j++){
			v[i*prime[j]]=true;
			if(i%prime[j]==0)break;
			mu[i*prime[j]]=-mu[i];
		}
	}
	for(int i=1;i<=maxn-10;i++)
	for(int j=i;j<=maxn-10;j+=i)buildroad(j,i);
}

int main()
{
	work();int Case=0;
	while(scanf("%d %d",&n,&m),n!=0&&m!=0)
	{
		printf("Case #%d:\n",++Case);
		memset(tr,0,sizeof(tr));
		for(int i=1,id,N,D,V;i<=m;i++)
		{
			scanf("%d",&id);
			if(id==1){
				scanf("%d %d %d",&N,&D,&V);
				if(N%D==0)for(int j=first[N/D];j;j=e[j].next)add(e[j].y*D,V*mu[e[j].y]);
			}else{
				scanf("%d",&N);ll ans=0;
				for(int l=1,r;l<=N;l=r+1){
					r=N/(N/l);
					ans+=N/l*(sum(r)-sum(l-1));
				}
				printf("%lld\n",ans);
			}
		}
	}
}