【狄利克雷卷积+快速幂】HDU5628[Clarke and math]题解

题目概述

给出数论函数 $f(n)$ 和 $k$ ，求函数 $g$ ，其中 $g(i)=\sum_{i_1|i}\sum_{i_2|i_1}\cdots\sum_{i_k|i_{k-1}}f(i_k)$ 。

解题报告

很容易发现是狄利克雷卷积的形式，我们令 $F_k(i_{k-1})=\sum_{i_k|i_{k-1}}f(i_k)$ ，即 $F_k=f*1$ 。以此类推， $F_{k-1}=F_{k}*1$ ， $F_{k-2}=F_{k-1}*1$ ……

最后得到 $g=f*1*1*\cdots*1$ ，因为狄利克雷卷积满足分配律和交换律，所以我们可以用快速幂。每次求卷积可以用类似埃氏筛的方法，效率 $O(nlog_2n)$ 。

示例程序

#include<cstdio>
using namespace std;
typedef long long LL;
const int maxn=100000,MOD=1e9+7;

int te,n,K;
struct F {int s[maxn+5];void unit() {s[1]=1;for (int i=2;i<=n;i++) s[i]=0;}};
F c,one,ans;

inline void AMOD(int &x,int tem) {if ((x+=tem)>=MOD) x-=MOD;}
F operator * (const F &a,const F &b)
{
    for (int i=1;i<=n;i++) c.s[i]=0;
    for (int i=1;i<=n;i++)
    for (int j=i;j<=n;j+=i)
        AMOD(c.s[j],(LL)a.s[i]*b.s[j/i]%MOD);
    return c;
}
F Pow(F w,int b)
{
    F s;s.unit();
    while (b) {if (b&1) s=s*w;b>>=1;if (b) w=w*w;}
    return s;
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    for (scanf("%d",&te);te;te--)
    {
        scanf("%d%d",&n,&K);
        for (int i=1;i<=n;i++) scanf("%d",&ans.s[i]),one.s[i]=1;ans=ans*Pow(one,K);
        for (int i=1;i<=n;i++) printf("%d",ans.s[i]),putchar(i==n?'\n':' ');
    }
    return 0;
}