已知 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}) H−G≡0(xn/2)
因为要变成 x n x^n xn,所以两边平方
对于 H 2 − 2 H G + G 2 H^2-2HG+G^2 H2−2HG+G2 化简,直接乘 F F F
最后可以推出: G = 2 H − F H 2 G=2H-FH^2 G=2H−FH2,递归然后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 的多项式完全输出应该是这样的: