P6222 简单题加强版——各种推式子trick大杂烩

Description

求$$\sum _{i=1}^n\sum _{j=1}^n(i+j{)}^t{\mu }^2(\gcd (i,j))\gcd (i,j)$$

$q$组询问，$t$为常量，$n$为变量。每个答案都要对$2^{32}$取模。$n\le 10^7,q\le 10^4$。

时间限制$0.5s-1.5s$，空间限制256MB

Solution

Part 1: 莫反推式子

一道比较套路的推式子题，中间用到了许多推式子的trick。

$$\sum _{i=1}^n\sum _{j=1}^n(i+j{)}^t{\mu }^2(\gcd (i,j))\gcd (i,j)$$

$$=\sum _{k=1}^n{\mu }^2(k)k\sum _{i=1}^n\sum _{j=1}^n[\gcd (i,j)=k](i+j{)}^t$$

$$=\sum _{k=1}^n{\mu }^2(k)k^{t+1}\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }[\gcd (i,j)=1](i+j{)}^t$$

$$=\sum _{k=1}^n{\mu }^2(k)k^{t+1}\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }(\sum _{d|gcd(i,j)}\mu (d))(i+j{)}^t$$

$$=\sum _{k=1}^n{\mu }^2(k)k^{t+1}\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)d^t(\sum _{i=1}^{\lfloor \frac{n}{kd}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{kd}\rfloor }(i+j{)}^t))$$

令$F(x)={\sum }_{i=1}^x{\sum }_{j=1}^x(i+j{)}^f$，带入得

$$\sum _{k=1}^n{\mu }^2(k)k^{t+1}\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)d^{t}F(\lfloor \frac{n}{kd}\rfloor )$$

令$T=kd$，转而枚举$T$并将枚举$T$的$\sum$提到最外层，

$$\sum _{T=1}^{n}F(\lfloor \frac{n}{T}\rfloor )\sum _{k|T}{\mu }^2(k)k^{t+1}\mu (\frac{T}{k})(\frac{T}{k}{)}^t$$

$$=\sum _{T=1}^{n}F(\lfloor \frac{n}{T}\rfloor )T^k\sum _{k|T}{\mu }^2(k)k\mu (\frac{T}{k})$$

令$G(T)=T^k{\sum }_{k|T}{\mu }^2(k)k\mu (\frac{T}{k})$，带入得

$$\sum _{T=1}^{n}F(\lfloor \frac{n}{T}\rfloor )G(T)$$

这不是卷积形式吗 可以发现这是一个整除分块的套路式。现在我们只需要快速求出单个$F$以及一个区间的$G$之和即可，而后者可以通过前缀和求出。所以现在我们只需要思考如何快速求出$F,G$函数。

Part 2: 求函数F

先考虑$F(n)$。

$$\sum _{i=1}^n\sum _{j=1}^n(i+j{)}^f$$

考虑去枚举$i+j$。对于一个$(i+j)=sum$它的贡献是$su{m}^k{p}_{sum}$，其中$p_{sum}$表示任选两个在$[1,n]$区间内的整数使得它们的和恰好为$sum$的方案数。

先考虑如何求出$p_{sum}$。分类讨论：
①若$sum\le n$，那么$p_{sum}=sum-1$；
②否则$p_{sum}=2n-sum+1$。

我们枚举$sum$(即下式中的$i$)：

$$\sum _{i=1}^n(i-1)i^k+\sum _{i=n+1}^{2n}(2n-i+1)i^k$$

$$=(\sum _{i=1}^n\sum _{j=i+1}^n{j}^k)+(\sum _{i=n+1}^{2n}\sum _{j=n+1}^i{j}^k)$$

令$S(x)={\sum }_{i=1}^x{i}^f$，带入得

$$(\sum _{i=1}^{n}S(n)-S(i))+\sum _{i=n+1}^{2n}S(i)-S(n)$$

$$=nS(n)-\sum _{i=1}^{n}S(i)+\sum _{i=n+1}^{2n}S(i)-nS(n)$$

$$=\sum _{i=n+1}^{2n}S(i)-\sum _{i=1}^{n}S(i)$$

$$=(\sum _{i=1}^{2n}S(i)-\sum _{i=1}^{n}S(i))-\sum _{i=1}^{n}S(i)$$

令$S^{\prime }(x)={\sum }_{i=1}^{x}S(i)$，带入得

$F(n)=S^{\prime }(2n)-2S(n)$

于是我们对$S$做两遍前缀和处理即可$O(1)$调用求出$F$。

注意$S$可以通过积性筛预处理求出来。

Part 3: 求函数G

观察一下:$$G(T)=\sum _{k|T}{\mu }^2(k)k\mu (\frac{T}{k})$$

可以发现，这个函数是由许多积性函数($id,\mu$)组成的。所以$G$本身也是一个积性函数，所以可以考虑线性筛积性函数。

线性筛积性函数只需要思考一种情况即可得到一般的筛法。这就是$T=p^c$的情况，其中$p$为质数且$c$为自然数。

①$c=0$: 此时$T=1$，$G(T)=1$；
②$c=1$: 此时

$$G(T)={\mu }^2(1)\times 1\times \mu (T)+{\mu }^2(T)\times T\times \mu (1)$$

即在这种情况中$G(T)=T-1$。

③$c=2$: 显然只有${\mu }^2(p)\times p\times \mu (p)$能产生非$0$的贡献，而这个值等于$1\times p\times (-1)=-p$，所以$G(T)=-p$

④当$c\ge 3$时，对于任意满足$u+v=c$的正整数对$(u,v)$都存在$u>1$或$v>1$，而任何质数的超过$1$次方的莫比乌斯函数值都是$0$，所以此时$G(T)=0$。

---

为什么说考虑了$T=p^c$的情况之后就完全搞定了这种函数的积性筛呢？我们看一下$G$的线性筛的代码：

//judge记录了这个数是否为合数
for (int i=2;i<=len;i++){
	if (!judge[i])  G[i]=i-1,prime[++pos]=i;
	//i是质数，对应的G值为i-1
	for (int j=1;j<=pos;j++){
		int p=prime[j];
		if (i*p>len)  break;//超过边界, 跳出
		judge[i*p]=1;//i*p为合数
		if (i%p==0){
			int q=i/p;
			if (q%p)  G[i*p]=-p*G[i/p];
			//核心语句,若不理解请先看下去
			break;
		}
		else G[i*p]=G[i]*G[p];//积性函数递推
	}
}

当$i$被$p$整除时，一定能分解成$i=p^c\times y$的形式，其中$c\ge 1$。即，$i\times p=p^{c^{\prime }}\times y$，其中$c^{\prime }\ge 2$。

由于$p$为质数，所以$p^{c^{\prime }}$与$y$互质，即$G(ip)=G(p^{c^{\prime }})\times G(y)$。$G(p^{c^{\prime }})$就是我们上面分类讨论的东西，$G(y)$是可以$O(1)$调用的东西。所以这样就可以完成线性筛。

Part 4: 空间复杂度与时间复杂度的评估

本题严重卡空间，所以开的数组必须很少，只需要开下面几个数组：

①$prime$(素数)集合，由于在$[1,2\times 10^7]$内的质数很少，所以这个集合可以少开一点或者直接使用$vector$；
②素数校验器$judge$，一定要开成bool；
③$G$函数与$i^k$函数。

不难发现，$F$函数可以直接调用$i^k$函数$O(1)$求出，不需要预处理，直接放在主函数中除了大了几倍时间常数之外并无大碍。

考虑这个空间是多少，并令$m=2\times 10^7$。
①$prime$: 开到$\frac{m}{2}$；
②③开到$m$。

由于$bool$占用的空间只有$int$的一半，所以总空间占用为$\frac{m}{2}+\frac{m}{2}+m+m=3m=6\times 10^7$。同时，我们不开$longlong$，采用unsigned int使其自然溢出实现取模即可防止$MLE$。

时间并没有卡得很厉害，只要复杂度正确并且开了$O2$提交就能过。

总时间复杂度$O(\frac{n}{\ln (n)}\log k+n+T\sqrt{n})$。

Summary

这是一道套路题，考察了推式子的能力与一些卡空间的trick。

本题先使用了莫比乌斯反演得到了一个简化的式子，将问题拆成了两个子问题: 求出函数$F$与$G$。在求$F$过程中我们巧妙枚举了$i+j$，并将系数转化得到简明扼要的式子；在求$G$过程中我们巧妙运用了积性函数的性质，使用分类讨论+线性筛求出了$G$。最终，我们卡了一波空间，苦尽甘来，正解出现。

Code

//by ducati
#include<bits/stdc++.h>
#define int unsigned int
using namespace std;
const int maxl=20005;

int t,len,k,n,pos=0,ans=0;
int prime[maxl/5],pre_k[maxl],f[maxl];
bool judge[maxl];

int quick_power(int x,int y){
	int res=1;
	for (;y;y=y>>1,x*=x)
	  if (y&1)  res*=x;
	return res;
}

void init(){
	pre_k[1]=1,f[1]=1;
	for (int i=2;i<=len;i++){
		if (!judge[i])  pre_k[i]=quick_power(i,k),f[i]=i-1,prime[++pos]=i;
		for (int j=1;j<=pos;j++){
			int p=prime[j];
			if (i*p>len)  break;
			pre_k[i*p]=pre_k[i]*pre_k[p];
			judge[i*p]=1;
			if (i%p==0){
				int q=i/p;
				if (q%p)  f[i*p]=-p*f[i/p];
				break;
			}
			else f[i*p]=f[i]*f[p];
		}
	}
	for (int i=1;i<=len;i++)  f[i]=f[i-1]+f[i]*pre_k[i];
	for (int i=1;i<=len;i++)  pre_k[i]+=pre_k[i-1];
	for (int i=1;i<=len;i++)  pre_k[i]+=pre_k[i-1];
}
int calc(int x){return pre_k[2*x]-2*pre_k[x];}

signed main(){
	cin>>t>>len>>k;len*=2;
	init();
	while (t--){
		cin>>n;ans=0;
		for (int l=1,r;l<=n;l++){
			r=n/(n/l);
			ans+=calc(n/l)*(f[r]-f[l-1]);
			l=r;
		}
		cout<<ans<<endl;
	}
	return 0;
}

sto Fee_clе6418 orz