[HDU5628][数学]Clarke and math

HDU5628

就是求狄利克雷卷积的k次幂
狄利克雷卷积满足结合律，所以可以快速幂，类似矩阵
那么就是要解决怎么乘的问题
枚举因数，看它会对哪些产生贡献就完了
时间复杂度$O(nlognlogk)$

Code：

#include<bits/stdc++.h>
#define mod 1000000007
#define ll long long
using namespace std;
inline int read(){
    int res=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
    while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
    return res*f;
}
inline ll add(ll x,ll y){x+=y;if(x>=mod) x-=mod;return x;}
inline ll dec(ll x,ll y){x-=y;if(x<0) x+=mod;return x;}
inline ll mul(ll x,ll y){return x*y%mod;}
inline void inc(ll &x,ll y){x+=y;if(x>=mod) x-=mod;}
const int N=1e5+5;
ll tmp[N];
int n,k;
inline void Mul(ll *a,ll *b){
    memset(tmp,0,sizeof(tmp));
    for(int i=1;i*i<=n;i++){
        inc(tmp[i*i],mul(a[i],b[i]));
        for(int j=i+1;i*j<=n;j++) inc(tmp[i*j],add(mul(a[i],b[j]),mul(a[j],b[i])));
    }
    memcpy(a,tmp,sizeof(tmp));
}
ll x[N],res[N],f[N];
int main(){
    int t=read();
    while(t--){
        n=read(),k=read();
        for(int i=1;i<=n;i++) f[i]=read();
        for(int i=1;i<=n;i++) x[i]=1,res[i]=0;
        res[1]=1;
        for(;k;k>>=1,Mul(x,x)) if(k&1) Mul(res,x);
        Mul(res,f);
        for(int i=1;i<n;i++) cout<<res[i]<<" ";
        cout<<res[n]<<"\n";
    }
    return 0;
}