【HDU5628】Clarke and math-狄利克雷卷积+快速幂

测试地址：Clarke and math
题目大意：设$f$和$g$是定义在集合$\{1,2,...,n\}$的一个函数，且$g(i)=\sum_{i_1|i}\sum_{i_2|i_1}...\sum_{i_k|i_{k-1}}f(i_k)$，给定$k$和$f(1),f(2),...f(n)$，求$g(1),g(2),...,g(n)$。
做法：此题需要用到狄利克雷卷积+快速幂。
定义两个算术函数$f$和$g$的狄利克雷卷积为：$(f*g)(n)=\sum_{d|n}f(d)g(n/d)$，那么函数的狄利克雷卷积满足以下性质：
交换律：$f*g=g*f$。
结合律：$f*(g*h)=(f*g)*h$。
分配律：$f*(g+h)=f*g+f*h$。
那么再看我们要求的函数$g$，我们一层一层展开，设函数$I(n)=1$，则：
$f'(i_{k-1})=\sum_{i_k|i_{k-1}}f(i_k)=\sum_{i_k|i_{k-1}}f(i_k)I(i_{k-1}/i_k)$，即$f'=I*f$
$f''(i_{k-2})=\sum_{i_{k-1}|i_{k-2}}f'(i_{k-1})=\sum_{i_{k-1}|i_{k-2}}f'(i_{k-1})I(i_{k-2}/i_{k-1})$，即$f''=I*f'=I*I*f$
$...$
这样一直推导下去，得到$g=I*I*I*...*I*f$（$k$个$I$），由于狄利克雷卷积满足结合律，所以可以用快速幂加速计算。在计算狄利克雷卷积的时候，如果对于每一个$1\le i\le n$都按照定义枚举约数计算贡献的话，时间复杂度显然爆炸，所以我们可以枚举约数，然后枚举有哪些$i$需要加上这个约数的贡献即可，那么计算一次狄利克雷卷积的复杂度为$O(n\log n)$，总的时间复杂度就是$O(n\log n\log k)$。
注意如果用结构体传参会爆栈，直接用数组+指针传参就可以了。以及这题的格式要求行末不能有空格，但要有换行，注意一下就行了。
狄利克雷卷积的作用当然不只是做这种题，它为一堆有关莫比乌斯反演和杜教筛的式子提供了简明的记号，这个后面有时间再进行小结，这里就不再赘述了。
以下是本人代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define mod 1000000007
#define ll long long
using namespace std;
int T,n,k;
ll f[100010],g[100010],s[100010],ss[100010],pro[100010];

void mult(ll *a,ll *b)
{
  memset(s,0,sizeof(s));
  for(int i=1;i*i<=n;i++)
  {
    s[i*i]=(s[i*i]+a[i]*b[i])%mod;
    for(int j=i+1;i*j<=n;j++)
      s[i*j]=(s[i*j]+a[i]*b[j]+a[j]*b[i])%mod;
  }
  for(int i=1;i<=n;i++) a[i]=s[i];
}

void power(ll *a,int p)
{
  for(int i=1;i<=n;i++) ss[i]=a[i];
  memset(pro,0,sizeof(pro));
  pro[1]=1;
  while(p)
  {
    if (p&1) mult(pro,ss);
    mult(ss,ss);p>>=1;
  }
  for(int i=1;i<=n;i++) a[i]=pro[i];
}

int main()
{
  scanf("%d",&T);
  while(T--)
  {
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
      scanf("%lld",&f[i]);
      g[i]=1;
    }
    power(g,k);
    mult(f,g);
    for(int i=1;i<=n;i++)
      printf("%lld%c",f[i],(i==n)?'\n':' ');
  }

  return 0;
}