一种快速的常系数齐次线性递推算法
论文参考
https://arxiv.org/pdf/2008.08822.pdf
首先常系数齐次线性递推问题可以转化为这样一个问题
给定多项式\(p(x)\)和多项式\(q(x)\),求\(\frac{p(x)}{q(x)}\)在\(n\)次项系数,\(n\)可能很大。
具体转化的话
大概是设递推式为\(f_n=\sum_{i=1}^{len} f_{n-i}*g_i\)
我们要做的就是求出这个递推式\(f_n\)的生成函数。
可以证明其可以表示为\(\frac{p(x)}{q(x)}\)的形式
其中\(q(x)=1-g(x)\),\(p(x)=f(x)*(1-g(x)) (mod \ x^{len})\)
证明不是很会。
我们考虑怎么求解\(\frac{p(x)}{q(x)}\)的第\(n\)项
\[\frac{f(x)}{g(x)}=\frac{p(x)*q(-x)}{q(x)q(-x)}
\]
令\(v(x)=q(x)q(-x)\)
有这样一个性质,\(v(x)=v(-x)\),所以$v(x)只有偶数次项有值
所以
\[\frac{p(x)}{q(x)}=\frac{L(x^2)}{U(x^2)}+x*\frac{R(x^2)}{U(x^2)}
\]
按照这个式子以及\(n\)的奇偶性向下递归即可。
int t[N],p[N],q[N],dp[N],dq[N],ddp[N],ddq[N];
int coefficient(int n,int len)
{
int v=inv(2),wn=ksm(h,(mo-1)/(2*len)),wm=inv(wn);
for(int i=0;i<len;i++)dp[i]=p[i],dq[i]=q[i];
ntt(dp,len,+1);ntt(dq,len,+1);
while(n)
{
//ddp ddq
for(int i=0;i<len;i++)ddp[2*i]=dp[i],ddq[2*i]=dq[i];
for(int i=0,w=1;i<len;i++,w=1ll*w*wn%mo)p[i]=1ll*p[i]*w%mo,q[i]=1ll*q[i]*w%mo;
ntt(p,len,+1);ntt(q,len,+1);
for(int i=0;i<len;i++)ddp[2*i+1]=p[i],ddq[2*i+1]=q[i];
//p dp
for(int i=0;i<2*len;i++)t[i]=1ll*ddp[i]*ddq[i^len]%mo;
if(n&1)for(int i=0,w=1;i<2*len;i++,w=1ll*w*wm%mo)t[i]=1ll*t[i]*w%mo;
for(int i=0;i<len;i++)t[i]=1ll*v*inc(t[i],t[i+len])%mo,dp[i]=t[i];
ntt(t,len,-1);for(int i=0;i<len;i++)p[i]=t[i];
//q dq
for(int i=0;i<2*len;i++)t[i]=1ll*ddq[i]*ddq[i^len]%mo;
for(int i=0;i<len;i++)t[i]=1ll*v*inc(t[i],t[i+len])%mo,dq[i]=t[i];
ntt(t,len,-1);for(int i=0;i<len;i++)q[i]=t[i];
n>>=1;
}
return 1ll*p[0]*inv(q[0])%mo;
}
int recurrence(int *f,int *g,int n,int k)
{
int len=1;
while(len<(n+1))len<<=1;
for(int i=0;i<len;i++)p[i]=q[i]=0;
q[0]=1;for(int i=1;i<=n;i++)q[i]=dec(0,g[i]);
for(int i=0;i<n;i++)a[i]=f[i];
for(int i=0;i<=n;i++)b[i]=q[i];
poly_mul(len,len);
for(int i=0;i<n;i++)p[i]=a[i];
return coefficient(k,len);
}