多项式求逆

来源:互联网 发布:php进销存管理系统 编辑:程序博客网 时间:2024/04/27 09:23

设已知A(x)B(x)=1(mod x^n),考虑A(x)C(x)=1(mod x^2n)

显然B(x)-C(x)=0(mod x^n)

平方得B(x)^2-2B(x)C(x)+C(x)^2=0(mod x^2n)

同乘A(x)得A(x)B(x)^2-2B(x)+C(x)=0(mod x^2n)

即有C(x)=2B(x)-A(x)B(x)^2,fft即可

设A(x)常数项为t,则A(x)*1/t=1(mod x^1)/

不断倍增即可解决

注意fft又是循环卷积,实现的时候必须做适当的清零

然后每个式子在不全相同的模意义下成立,清零的区间是哪段也要注意一下

代码:

#include<cstdio>namespace poly{#define mo 998244353struct AwD{int x;};AwD operator+(AwD a,AwD b){return (AwD){(a.x+b.x)%mo};}AwD operator-(AwD a,AwD b){return (AwD){(a.x-b.x+mo)%mo};}AwD operator*(AwD a,AwD b){return (AwD){(int)(1LL*a.x*b.x%mo)};}AwD operator^(AwD a,int b){if(b<0) b+=mo-1;if(!b) return (AwD){1};AwD temp=a^(b>>1);temp=temp*temp;if(b&1) temp=temp*a;return temp;}AwD operator/(AwD a,AwD b){return a*(b^-1);}const AwD root=(AwD){3};const int om=mo-1;void ntt(AwD*a,int n,int d){int i,j,k;AwD w,t,u,v;for(i=(n>>1),j=1;j<n;j++){if(i<j) t=a[i],a[i]=a[j],a[j]=t;for(k=(n>>1);i&k;i^=k,k>>=1);i^=k;}for(k=2;k<=n;k<<=1){w=root^((mo-1)/k*d);for(i=0;i<n;i+=k){t=(AwD){1};for(j=i;j<i+(k>>1);j++){u=a[j];v=t*a[j+(k>>1)];a[j]=u+v;a[j+(k>>1)]=u-v;t=t*w;}}}}AwD a[1<<20],b[1<<20];void print(AwD*a,int l){for(int i=0;i<l;i++) printf("%d ",a[i].x);printf("\n");}void plus(AwD*_a,AwD*_b,int l,AwD*c){for(int i=0;i<l;i++) a[i]=_a[i],b[i]=_b[i];for(int i=0;i<l;i++) c[i]=a[i]+b[i];}void subt(AwD*_a,AwD*_b,int l,AwD*c){for(int i=0;i<l;i++) a[i]=_a[i],b[i]=_b[i];for(int i=0;i<l;i++) c[i]=a[i]-b[i];}void mult(AwD*_a,AwD b,int l,AwD*c){for(int i=0;i<l;i++) a[i]=_a[i];for(int i=0;i<l;i++) c[i]=a[i]*b;}void mult(AwD*_a,AwD*_b,int l,AwD*c){for(int i=0;i<l;i++) a[i]=_a[i],b[i]=_b[i];ntt(a,l,1);ntt(b,l,1);for(int i=0;i<l;i++) c[i]=a[i]*b[i];ntt(c,l,-1);for(int i=0;i<l;i++) c[i]=c[i]/(AwD){l};}AwD a1[1<<20],aa[1<<20],tmp[1<<20];void inv(AwD*_a,int l,AwD*b){for(int i=0;i<l;i++) a1[i]=_a[i];for(int i=0;i<l;i++) b[i]=i?(AwD){0}:a1[i]^-1;for(int l0=2;l0<=l;l0<<=1){mult(b,(AwD){2},l0>>1,tmp);mult(b,b,l0,b);for(int i=0;i<(l0<<1);i++) aa[i]=i<l0?a1[i]:(AwD){0};mult(aa,b,l0<<1,b);for(int i=l0;i<(l0<<1);i++) b[i]=(AwD){0};subt(tmp,b,l0,b);}}}int n,l;poly::AwD a[1<<20];int main(){scanf("%d",&n);n++;for(int i=0;i<n;i++) scanf("%d",&a[i].x);l=1;while(l<n) l<<=1;for(int i=n;i<l;i++) a[i].x=0;poly::inv(a,l,a);for(int i=0;i<n;i++) printf("%d ",a[i].x);}

0 0