[学习笔记]多项式的各种操作

目录

• 多项式求逆
• 多项式除法/取模
• 多项式牛顿迭代法
• 多项式开根
• 多项式 ln
• 多项式 exp
• 多项式幂函数

多项式求逆

点此看题

可以暴力递推，时间复杂度$O(n^2)$。

倍增可以将之优化到$O(n\log n)$，我们先从$1$开始倍增，假设现在倍增到了$p$，得到了模$x^p$的逆元$B^{\prime }(x)$，现在我们要求$B(x)$，显然这时候求出来的$B^{\prime }(x)$是$B(x)$的一部分，有下列关系式：
$$B(x)-B^{\prime }(x)=0\mathrm{m}\mathrm{o}\mathrm{d}{x}^p$$考虑$B(x)-B^{\prime }(x)$的结果一定是形如$0+0x+0x^2....+a_p{x}^p....$，怎么把模数扩展到$2p$呢，可以考虑将它平方，那么平方过的结果形如$0+0x+0x^2....+0x^{2p-1}+a_{2p}{x}^{2p}$，有如下关系式：
$$(B(x)-B^{\prime }(x){)}^2=0\mathrm{m}\mathrm{o}\mathrm{d}{x}^{2p}$$$$B(x{)}^2+B^{\prime }(x{)}^2-2B(x)B^{\prime }(x)=0\mathrm{m}\mathrm{o}\mathrm{d}{x}^{2p}$$$$B(x)+B^{\prime }(x{)}^{2}A(x)-2B^{\prime }(x)=0\mathrm{m}\mathrm{o}\mathrm{d}{x}^{2p}$$$$B(x)=2B^{\prime }(x)-B^{\prime }(x{)}^{2}A(x)\mathrm{m}\mathrm{o}\mathrm{d}{x}^{2p}$$然后就可以用卷积算了，时间复杂度$O(n\log n+\frac{n}{2}\log \frac{n}{2}.....)=O(n\log n)$。

#include <cstdio>
#include <iostream>
#include <cmath>
#define int long long
using namespace std;
const int MAXN = 1000005;
const int MOD = 998244353;
int read()
{
   
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,len,cur,lg,a[MAXN],b[MAXN],A[MAXN],B[MAXN],Rev[MAXN];
int qkpow(int a,int b)
{
   
	int res=1;
	while(b>0)
	{
   
		if(b&1) res=res*a%MOD;
		a=a*a%MOD;
		b>>=1;
	}
	return res;
}
void NTT(int *a,int len,int tmp)
{
   
	lg=(int)round(log2(len));
	for(int i=0;i<len;i++)
	{
   
		Rev[i]=(Rev[i>>1]>>1)|((i&1)<<(lg-1));
		if(i<Rev[i])
			swap(a[i],a[Rev[i]]);
	}
	for(int s=2;s<=len;s<<=1)
	{
   
		int t=s/2,w=(tmp==1)?qkpow(3,(MOD-1)/s):qkpow(3,(MOD-1)-(MOD-1)/s);
		for(int i=0;i<len;i+=s)
		{
   
			int x=1;
			for(int j=0;j<t;j++,x=x*w%MOD)
			{
   
				int fe=a[i+j],fo=a[i+j+t];
				a[i+j]=(fe+x*fo)%MOD;
				a[i+j+t]=((fe-fo*x)%MOD+MOD)%MOD;
			}
		}
	}
	if(tmp==1) return ;
	int inv=qkpow(len,MOD-2);
	for(int i=0;i<len;i++)
		a[i]=a[i]*inv%MOD;
}
void work(int n)
{
   
	len=1;while(len<=2*n) len<<=1;
	for(int i=0;i<len;i++) A[i]=B[i]=0;
	for(int i=0;i<n;i++) A[i]=a[i];
	for(int i=0;i<n;i++) B[i]=b[i];
	NTT(A,len,1);NTT(B,len,1);
	for(int i=0;i<len;i++) A[i]=2*B[i]-B[i]*B[i]%MOD*A[i],A[i]%=MOD;
	NTT(A,len,-1);
	for(int i=0;i<n;i++) b[i]=(A[i]%MOD+MOD)%MOD;
}
signed main()
{
   
	n=read();
	for(int i=0;i<n;i++) a[i]=read();
	b[0]=qkpow(a[0],MOD-2);
	cur=1;
	while(cur<n)
	{
   
		cur<<=1;
		work(cur);
	}
	for(int i=0;i<n;i++)
		printf("%lld ",b[i]);
}

多项式除法

点此看题

给定$n$次多项式$A(x)$，$m$次多项式$B(x)$（$n>m$），求出$n-m$次的商$C(x)$和$m-1$次的余数$D(x)$

由于这里有余数$D(x)$难于处理，我们可以考虑消去它的影响。

可以用一些奇技淫巧，比如说翻转多项式，我们来推一波柿子（$R$表示反转上标）：
$$A(\frac{1}{x})=B(\frac{1}{x})C(\frac{1}{x})+D(\frac{1}{x})$$$$x^{n}A(\frac{1}{x})=x^{m}B(\frac{1}{x})\cdot x^{n-m}C(\frac{1}{x})+x^{n}D(\frac{1}{x})$$ A R ( x ) = B R ( x ) ⋅ C R ( x ) + x n