题目
https://www.luogu.com.cn/problem/P4726
思路
好的,相信大家对牛顿迭代了解已经很透彻了,这里还是给出结论:
若
G
(
F
(
x
)
)
≡
0
(
m
o
d
x
n
)
G(F(x))\equiv0\ (mod \ x^n)
G(F(x))≡0 (mod xn) 其中
G
(
x
)
G(x)
G(x) 是给定的多项式,那么
F
(
x
)
≡
F
0
(
x
)
−
G
(
F
0
(
x
)
)
G
′
(
F
0
(
x
)
)
(
m
o
d
x
n
)
F(x)\equiv F_0(x)-\frac{G(F_0(x))}{G'(F_0(x))}\ (mod \ x^n)
F(x)≡F0(x)−G′(F0(x))G(F0(x)) (mod xn) 其中
F
0
(
x
)
F_0(x)
F0(x) 为
F
(
x
)
F(x)
F(x) 在
m
o
d
x
n
2
mod x^{\frac{n}{2}}
modx2n 意义下的多项式。
好的我们现在开始exp
F
(
x
)
≡
e
A
(
x
)
(
m
o
d
x
n
)
F(x)\equiv e^{A(x)}\ (mod\ x^n)
F(x)≡eA(x) (mod xn)
化简一下得
l
n
(
F
(
x
)
)
−
A
(
x
)
≡
0
(
m
o
d
x
n
)
ln(F(x))-A(x)\equiv 0\ (mod\ x^n)
ln(F(x))−A(x)≡0 (mod xn)
令
G
(
F
(
x
)
)
=
l
n
(
F
(
x
)
)
−
A
(
x
)
G(F(x))=ln(F(x))-A(x)
G(F(x))=ln(F(x))−A(x)
则
G
′
(
F
(
x
)
)
=
1
F
(
x
)
G'(F(x))=\frac{1}{F(x)}
G′(F(x))=F(x)1
(我也不知道为什么A(x)可以看成常数)
于是我们套牛顿迭代公式得: F ( x ) ≡ F 0 ( x ) − G ( F 0 ( x ) ) G ′ ( F 0 ( x ) ) ( m o d x n ) F(x)\equiv F_0(x)-\frac{G(F_0(x))}{G'(F_0(x))}\ (mod \ x^n) F(x)≡F0(x)−G′(F0(x))G(F0(x)) (mod xn) F ( x ) ≡ F 0 ( x ) ( 1 − l n ( F 0 ( x ) ) + A ( x ) ) ( m o d x n ) F(x)\equiv F_0(x)(1-ln(F_0(x))+A(x))\ (mod \ x^n) F(x)≡F0(x)(1−ln(F0(x))+A(x)) (mod xn)
然后递归求解就完啦
注意:该清空的数组要清空(多项式ln里面的求导数组和多项式逆数组)
代码
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=4e5+77,mod=998244353;
const ll g=3;
int r[N];
ll a[N],b[N],c[N],ia[N],da[N],_inv[N],f[N];
ll power(ll x,ll t)
{
ll b=1;
while(t)
{
if(t&1) b=b*x%mod; x=x*x%mod; t>>=1;
}
return b;
}
void dft(ll *a,int n,int f)
{
for(int i=1; i<=n; i++) if(i<r[i]) swap(a[i],a[r[i]]);
for(int i=2; i<=n; i<<=1)
{
int now=i>>1; ll wn=power(g,(mod-1)/i);
if(f==-1) wn=power(wn,mod-2);
for(int j=0; j<n; j+=i)
{
ll x,y,w=1;
for(int k=j; k<j+now; k++,w=w*wn%mod)
{
x=a[k],y=w*a[k+now]%mod;
a[k]=(x+y)%mod,a[k+now]=(x-y+mod)%mod;
}
}
}
if(f==-1) for(int i=0; i<n; i++) a[i]=a[i]*power(n,mod-2)%mod;
}
void inv(int n,ll *a,ll *b)
{
if(n==1)
{
b[0]=power(a[0],mod-2); return;
}
inv((n+1)>>1,a,b);
int len=1,lg=0;
while(len<n*2) len<<=1,lg++;
for(int i=1; i<=len; i++) r[i]=((r[i>>1]>>1)|((i&1)<<(lg-1)));
for(int i=0; i<n; i++) c[i]=a[i]; for(int i=n; i<len; i++) c[i]=0;
dft(c,len,1),dft(b,len,1);
for(int i=0; i<len; i++) b[i]=(2-c[i]*b[i]%mod+mod)%mod*b[i]%mod;
dft(b,len,-1);
for(int i=n; i<len; i++) b[i]=0;
}
void dev(int n,ll *a,ll *b)
{
for(int i=1; i<n; i++) b[i-1]=i*a[i]%mod; b[n-1]=0;
}
void invdev(int n,ll *a,ll *b)
{
for(int i=1; i<n; i++) b[i]=a[i-1]*_inv[i]%mod; b[0]=0;
}
void ln(int n,ll *a,ll *b)
{
dev(n,a,da); inv(n,a,ia);
int len=1,lg=0;
while(len<n*2) len<<=1,lg++;
for(int i=1; i<=len; i++) r[i]=((r[i>>1]>>1)|((i&1)<<(lg-1)));
dft(da,len,1); dft(ia,len,1);
for(int i=0; i<len; i++) c[i]=da[i]*ia[i]%mod;
dft(c,len,-1);
invdev(len,c,b);
for(int i=0; i<len; i++) da[i]=ia[i]=0;
}
void exp(int n,ll *a,ll *b)
{
if(n==1)
{
b[0]=1; return;
}
exp(n>>1,a,b);
ln(n,b,f);
f[0]=(a[0]+1-f[0]+mod)%mod;
for(int i=1; i<n; i++) f[i]=(a[i]-f[i]+mod)%mod;
int len=1,lg=0;
while(len<n<<1) len<<=1,lg++;
for(int i=1; i<=len; i++) r[i]=(r[i>>1]>>1)|((i&1)<<(lg-1));
dft(b,len,1),dft(f,len,1);
for(int i=0; i<len; i++) b[i]=b[i]*f[i]%mod;
dft(b,len,-1);
for(int i=n; i<len; i++) b[i]=f[i]=0;
// for(int i=0; i<n; i++) printf("%lld ",b[i]); printf("\n");
}
int main()
{
int n;
scanf("%d",&n);
for(int i=0; i<n; i++) scanf("%lld",&a[i]);
int len=1;
while(len<=n) len<<=1;
_inv[1]=1;
for(int i=2; i<=len; i++) _inv[i]=(mod-(mod/i))*_inv[mod%i]%mod;
exp(len,a,b);
for(int i=0; i<n; i++) printf("%lld ",b[i]);
}