多项式求逆

多项式求逆

板子 P4238

已知 A ( x ) A(x) A(x),求使得 A ( x ) B ( x ) ≡ 1 ( m o d x n ) A(x)B(x)\equiv 1 \pmod{x^n} A(x)B(x)1(modxn) B ( x ) B(x) B(x)
考虑倍增。
假设当前已得到 B ′ ( x ) B'(x) B(x),满足 A ( x ) B ′ ( x ) ≡ 1 ( m o d x n ) A(x)B'(x)\equiv 1 \pmod{x^n} A(x)B(x)1(modxn)
A ( x ) B ′ ( x ) − 1 ≡ 0 ( m o d x n 2 ) ( A ( x ) B ′ ( x ) − 1 ) 2 ≡ 0 ( m o d 2 n ) A ( x ) 2 B ′ ( x ) 2 − 2 A ( x ) B ′ ( x ) + 1 ≡ 0 ( m o d x n ) A ( x ) 2 B ′ ( x ) 2 − 2 A ( x ) B ′ ( x ) + A ( x ) B ( x ) ≡ 0 ( m o d x n ) B ( x ) ≡ 2 B ′ ( x ) − A ( x ) B ′ ( x ) 2 ( m o d x n ) A(x)B'(x)-1\equiv0\pmod{x^{\frac{n}{2}}}\\ (A(x)B'(x)-1)^2\equiv0\pmod{2^n}\\ A(x)^2B'(x)^2-2A(x)B'(x)+1\equiv0\pmod{x^n}\\ A(x)^2B'(x)^2-2A(x)B'(x)+A(x)B(x)\equiv0\pmod{x^n}\\ B(x)\equiv{2B'(x)-A(x)B'(x)^2}\pmod{x^n} A(x)B(x)10(modx2n)(A(x)B(x)1)20(mod2n)A(x)2B(x)22A(x)B(x)+10(modxn)A(x)2B(x)22A(x)B(x)+A(x)B(x)0(modxn)B(x)2B(x)A(x)B(x)2(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];
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;
}
signed main(){
	n=read();
	for(int i=0;i<n;i++) a[i]=read();
	getinv(a,b,n);
	for(int i=0;i<n;i++) printf("%d ",b[i]);
	printf("\n"); 
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值