多项式开根
已知
A
(
x
)
A(x)
A(x),求使得
B
(
x
)
2
≡
A
(
x
)
(
m
o
d
x
n
)
B(x)^2\equiv{A(x)\pmod{x^n}}
B(x)2≡A(x)(modxn)的
B
(
x
)
B(x)
B(x)
类似求逆,考虑倍增。
假设当前已知
B
′
(
x
)
2
≡
A
(
x
)
(
m
o
d
x
n
2
)
B'(x)^2\equiv{A(x)}\pmod{x^{\frac{n}{2}}}
B′(x)2≡A(x)(modx2n)
(
B
′
(
x
)
2
−
A
(
x
)
)
2
≡
0
(
m
o
d
x
n
)
B
′
(
x
)
4
−
2
B
′
(
x
)
2
A
(
x
)
+
A
(
x
)
2
≡
0
(
m
o
d
x
n
)
B
′
(
x
)
4
+
2
B
′
(
x
)
2
A
(
x
)
+
A
(
x
)
2
≡
4
A
(
x
)
B
′
(
x
)
2
(
m
o
d
x
n
)
(
B
′
(
x
)
2
+
A
(
x
)
)
2
≡
(
2
B
′
(
x
)
)
2
B
(
x
)
2
(
m
o
d
x
n
)
B
′
(
x
)
2
+
A
(
x
)
2
B
′
(
x
)
≡
B
(
x
)
(
m
o
d
x
n
)
(B'(x)^2-A(x))^2\equiv0\pmod{x^n}\\ B'(x)^4-2B'(x)^2A(x)+A(x)^2\equiv0\pmod{x^n}\\ B'(x)^4+2B'(x)^2A(x)+A(x)^2\equiv{4A(x)B'(x)^2}\pmod{x^n}\\ (B'(x)^2+A(x))^2\equiv{(2B'(x))^2B(x)^2}\pmod{x^n}\\ \frac{B'(x)^2+A(x)}{2B'(x)}\equiv{B(x)}\pmod{x^n}
(B′(x)2−A(x))2≡0(modxn)B′(x)4−2B′(x)2A(x)+A(x)2≡0(modxn)B′(x)4+2B′(x)2A(x)+A(x)2≡4A(x)B′(x)2(modxn)(B′(x)2+A(x))2≡(2B′(x))2B(x)2(modxn)2B′(x)B′(x)2+A(x)≡B(x)(modxn)
于是每次倍增向上赋值
代码
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int g=3;
const int mod=998244353;
const int M=2100009;
int read(){
int f=1,re=0;char ch;
for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
if(ch=='-'){f=-1,ch=getchar();}
for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
return re*f;
}
int ksm(int a,int b){
int ans=1;
while(b){
if(b&1) ans=(ll)ans*a%mod;
a=(ll)a*a%mod;
b>>=1;
}return ans%mod;
}
int n,r[M],tmp[M],a[M],b[M],cpy[M],F[M],inv2;
void ntt(int *A,int lim,int type){
for(int i=0;i<lim;i++) if(i<r[i]) swap(A[i],A[r[i]]);
for(int mid=1;mid<lim;mid<<=1){
int W=ksm(g,(mod-1)/(mid<<1));
for(int R=mid<<1,j=0;j<lim;j+=R){
int w=1;
for(ll k=0;k<mid;k++,w=(ll)w*W%mod){
int x=A[j+k],y=(ll)w*A[j+k+mid]%mod;
A[j+k]=(x+y)%mod;
A[j+mid+k]=(x-y+mod)%mod;
}
}
}
if(type==-1){
reverse(A+1,A+lim);
int inv=ksm(lim,mod-2);
for(int i=0;i<lim;i++) A[i]=(ll)A[i]*inv%mod;
}
}
void getinv(int a[],int b[],int len){
if(len==1){
b[0]=ksm(a[0],mod-2);
return;
}getinv(a,b,(len+1)>>1);
int lim=1,l=0;
while(lim<len+len) lim<<=1,l++;
for(int i=0;i<lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
for(int i=0;i<len;i++) tmp[i]=a[i];
for(int i=len;i<lim;i++) tmp[i]=0;
ntt(tmp,lim,1),ntt(b,lim,1);
for(int i=0;i<lim;i++) b[i]=(ll)b[i]*((2-(ll)tmp[i]*b[i]%mod+mod)%mod)%mod;
ntt(b,lim,-1);
for(int i=len;i<lim;i++) b[i]=0;
}
void getsqrt(int a[],int b[],int len){
if(len==1){
b[0]=1;
return;
}getsqrt(a,b,(len+1)>>1);
for(int i=0;i<=(len<<1);i++) F[i]=0;
getinv(b,F,len);
int lim=1,l=0;
while(lim<len+len) lim<<=1,l++;
for(int i=0;i<lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
for(int i=0;i<len;i++) cpy[i]=a[i];
for(int i=len;i<lim;i++) cpy[i]=0;
ntt(cpy,lim,1),ntt(b,lim,1),ntt(F,lim,1);
for(int i=0;i<lim;i++) b[i]=((ll)((ll)(b[i]+(ll)cpy[i]*F[i]%mod)%mod)*inv2)%mod;
ntt(b,lim,-1);
for(int i=len;i<lim;i++) b[i]=0;
}
signed main(){
n=read();inv2=ksm(2,mod-2);
for(int i=0;i<n;i++) a[i]=read();
getsqrt(a,b,n);
for(int i=0;i<n;i++) printf("%d ",b[i]);
printf("\n");
return 0;
}