多项式求逆

在这里插入图片描述
已知 F F F,求 G G G

考虑倍增
F ( x ) ∗ H ( x ) ≡ 1 ( m o d x n / 2 ) F(x) * H(x) \equiv 1 \pmod{x^{n/2}} F(x)H(x)1(modxn/2)

F ( x ) ∗ G ( x ) ≡ 1 ( m o d x n / 2 ) F(x) * G(x) \equiv 1 \pmod{x^{n/2}} F(x)G(x)1(modxn/2)

假设 H H H 已知,求G

做差可得: H − G ≡ 0 ( x n / 2 ) H-G\equiv 0(x^{n/2}) HG0(xn/2)

因为要变成 x n x^n xn,所以两边平方

对于 H 2 − 2 H G + G 2 H^2-2HG+G^2 H22HG+G2 化简,直接乘 F F F

最后可以推出: G = 2 H − F H 2 G=2H-FH^2 G=2HFH2,递归然后NTT即可

void Inv(int *F, int *G, int n) {
	if(n==1) {
		G[0]=pw(F[0], mo-2); //只有一项直接取逆元 
		return ; 
	}
	Inv(F, G, (n+1)>>1); 
	m=n; l=0; 
	for(n=1; n<(m<<1); n<<=1) ++l; //建议弄到2*m 
	for(i=0; i<n; ++i) r[i]=((r[i>>1]>>1) | ((i&1)<<l-1)); 
	for(i=0; i<m; ++i) c[i]=F[i]; 
	for(i=m; i<n; ++i) c[i]=0; //记得清0 
	NTT(c, 1, n); NTT(G, 1, n); 
	for(i=0; i<n; ++i) G[i]=Mod(2*G[i]%mo-c[i]*G[i]%mo*G[i]%mo); 
	NTT(G, -1, n); 
	for(i=m; i<n; ++i) G[i]=0; //记得清0 
}

注:

对于 n = 5 n=5 n=5 的多项式完全输出应该是这样的:

在这里插入图片描述

你可能感兴趣的:(多项式,NTT,多项式求逆)