hdu5628 Clarke and math

Problem

HDU链接

一句话题意

给定f，求g，我们定义

$$g(i)=\sum_{i_1|i}\sum_{i_2|i_1}\sum_{i_3|i_2}…\sum_{i_k|i_{k-1}}f(i_k)\bmod 1000000007$$

Solution

用Dirichlet卷积进行优化，Dirichlet卷积的定义是这样的

$$(f*g)(n)=\sum_{d|n}f(d)g(\frac n d)$$
观察由g到f的计算式，不难看出就是每一层求和就是一次迭代。
我们令 $I$为单位元。
当k=1时，我们有
$$g_1(i)=\sum_{d|i}f(d)*1\Leftrightarrow g_1=(f*I)$$
同样的，当k=2时，我们有
$$g_2(i)=\sum_{j|i}\sum_{d|j}f(d)*1=\sum_{j|i}g_1(j)*1\Leftrightarrow g_2=(I*(f*I))$$

所以题中所求就相当于

$$g_k=(f*I^k)$$
行末不要输出多余的空格，会卡格式！

Code

#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn=100010,mod=1000000007;
int z,n,k,f[maxn],g[maxn],tmp[maxn];
inline void pls (int &x){if(x>=mod) x-=mod;}
void mul(int *a,int *b)
{
    for(int i=1;i<=n;i++) tmp[i]=0;
    for(int i=1;i<=n;i++)
      for(int j=1;i*j<=n;j++)
        pls(tmp[i*j]+=(ll)a[i]*b[j]%mod);
    for(int i=1;i<=n;i++) a[i]=tmp[i];
}
int main()
{
    scanf("%d",&z);
    while(z--)
    {
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++) scanf("%d",&g[i]),f[i]=1;
        for(;k;k>>=1,mul(f,f))
          if(k&1)
            mul(g,f);
        for(int i=1;i<n;i++) printf("%d ",g[i]);
        printf("%d\n",g[n]);
    }
    return 0;
}