【HDU 5628】Clarke and math

传送门

---

problem

给出 $n,k$ 和数论函数 $f(i),i\in [1,n]$。对于 $\forall i\in [1,n]$，请求出：

$$g(i)=\sum _{i_1|i}\sum _{i_2|i_1}\sum _{i_3|i_2}\cdots \sum _{i_k|i_{k-1}}f(i_k)$$

答案对 $10^9+7$ 取模。

数据范围：$1\le n,k\le 10^5$，$0\le f(i)<10^9+7$。

---

solution

先介绍一下狄利克雷卷积，对于两个数论函数 $f(n)$ 和 $g(n)$，有：

$$(f\ast g)(n)=\sum _{d|n}f(d)g(\frac{n}{d})$$

求卷积的话，我们可以枚举约数 $d$，然后在倍数的地方加上相应的贡献，复杂度是 $O(n\log n)$ 的。

然后回到这道题来。

当 $k=1$ 的时候，有：

$$g_1(i)=\sum _{i_1|i}f(i_1)=\sum _{i_1|i}f(i_1)1(\frac{i}{i_1})=f\ast 1$$

其中 $\forall i\in N^{\ast },1(i)=1$。

当 $k=2$ 时，有：

$$g_2(i)=\sum _{i_1|i}{g}_1(i_1)=\sum _{i_1|i}{g}_1(i_1)1(\frac{i}{i_1})=g_1\ast 1=f\ast 1\ast 1$$

那么不断递推，发现最终的答案就是 $f\ast 1\ast 1\ast \cdots \ast 1$（就是 $f$ 和 $k$ 个 $1$ 卷起来）。

那么可以用快速幂优化这一过程，时间复杂度 $O(n{\log }^{2}n)。$

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define P 1000000007
using namespace std;
int n,k,F[N],I[N],ans[N],temp[N];
int add(int x,int y)  {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y)  {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y)  {return 1ll*x*y%P;}
void Dirichlet(int *f,int *g){
	memset(temp,0,sizeof(temp));
	for(int i=1;i*i<=n;++i){
		temp[i*i]=add(temp[i*i],mul(f[i],g[i]));
		for(int j=i+1;i*j<=n;++j)
			temp[i*j]=add(temp[i*j],add(mul(f[i],g[j]),mul(f[j],g[i])));
	}
	memcpy(f,temp,sizeof(temp));
}
void Solve(){
	for(int i=1;i<=n;++i)  I[i]=1,ans[i]=0;
	ans[1]=1;
	for(;k;k>>=1,Dirichlet(I,I))
		if(k&1)  Dirichlet(ans,I);
	Dirichlet(ans,F);
}
int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		scanf("%d%d",&n,&k);
		for(int i=1;i<=n;++i)  scanf("%d",&F[i]);
		Solve();
		for(int i=1;i<=n;++i)  printf("%d%c",ans[i],i==n?'\n':' ');
	}
	return 0;
}