巨佬制作人们大家好,我是练习多项式两周半的个人练习生lgl。这里总结一下多项式基本操作。
1.多项式加、减、输出
不说了。
时间复杂度$O(n)$。
2.多项式取模
已知多项式$F(x)$,求它对$x^n$取模。
人话:把$n$次及以上的系数清零。
时间复杂度$O(n)$。
3.多项式乘法/卷积
洛谷传送门。
(1)$FFT$
依靠复平面上瞎转以及强大的$double$。
#include<cmath> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; #define N 2000050 #define ll long long const double Pi = acos(-1.0); template<typename T> inline void read(T&x) {T f=1,c=0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=10*c+ch-'0';ch=getchar();}x = f*c; } struct cp {double x,y;cp(){}cp(double x,double y):x(x),y(y){}cp operator + (const cp&a)const{return cp(x+a.x,y+a.y);}cp operator - (const cp&a)const{return cp(x-a.x,y-a.y);}cp operator * (const cp&a)const{return cp(x*a.x-y*a.y,x*a.y+y*a.x);} }; int n,m,mx,to[2*N],lim=1,L; void fft(cp *a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){cp w0(cos(Pi/i),k*sin(Pi/i));for(int j=0;j<len;j+=(i<<1)){cp w(1,0);for(int o=0;o<i;o++,w=w*w0){cp w1 = a[j+o],w2 = a[j+o+i]*w;a[j+o] = w1+w2;a[j+o+i] = w1-w2;}}} } cp a[2*N],b[2*N],c[2*N]; int main() {read(n),read(m);mx = max(n,m);for(int i=0;i<=n;i++)read(a[i].x);for(int i=0;i<=m;i++)read(b[i].x);while(lim<=2*mx)lim<<=1,L++;for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1)));fft(a,lim,1),fft(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i];fft(c,lim,-1);for(int i=0;i<=n+m;i++)printf("%lld ",(ll)(c[i].x/lim+0.5));puts("");return 0; }
注意那个重载运算符,写在里面是可以连加连乘的。
(2)$NTT$
需要一个好模数,还要依靠原根的优秀性质。
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; #define N 2000050 #define ll long long #define MOD 998244353 template<typename T> inline void read(T&x) {T f=1,c=0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=10*c+ch-'0';ch=getchar();}x = f*c; } ll fastpow(ll x,int y) {ll ret = 1;while(y){if(y&1)ret=ret*x%MOD;x=x*x%MOD;y>>=1;}return ret; } int n,m,mx,to[2*N],lim=1,l; void ntt(ll *a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){ll w0 = fastpow(3,(MOD-1)/(i<<1));for(int j=0;j<len;j+=(i<<1)){ll w = 1;for(int o=0;o<i;o++,w=w*w0%MOD){ll w1 = a[j+o],w2 = a[j+o+i]*w%MOD;a[j+o] = (w1+w2)%MOD;a[j+o+i] = ((w1-w2)%MOD+MOD)%MOD;}}}if(k==-1)for(int i=1;i<(lim>>1);i++)swap(c[i],c[lim-i]); } ll a[2*N],b[2*N],c[2*N]; int main() {read(n),read(m);mx = max(n,m);for(int i=0;i<=n;i++)read(a[i]);for(int i=0;i<=m;i++)read(b[i]);while(lim<=2*mx)lim<<=1,l++;for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(l-1)));ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD;ntt(c,lim,-1);ll inv = fastpow(lim,MOD-2);for(int i=0;i<lim;i++)c[i]=c[i]*inv%MOD;for(int i=0;i<=n+m;i++)printf("%lld ",c[i]);puts("");return 0; }
自测不开$O2$时$NTT$较快,开了$O2$之后$FFT$更快。
(3)任意模数$NTT$
我的版本是用$FFT$+$long\;double$把系数拆成$x/M$和$x\;mod\;M$,之后再合并。
#include<cmath> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; #define double long double typedef long long ll; const int N = 500050; const double Pi = acos(-1.0); template<typename T> inline void read(T&x) {T f = 1,c = 0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}x = f*c; } int n,m,MOD; struct cp {double x,y;cp(){}cp(double x,double y):x(x),y(y){}cp operator + (const cp&a)const{return cp(x+a.x,y+a.y);}cp operator - (const cp&a)const{return cp(x-a.x,y-a.y);}cp operator * (const cp&a)const{return cp(x*a.x-y*a.y,x*a.y+y*a.x);} }; int to[N],lim,L; void init() {lim = 1,L = 0;while(lim<=2*max(n,m))lim<<=1,L++;for(int i=1;i<lim;i++)to[i] = ((to[i>>1]>>1)|((i&1)<<(L-1))); } ll A[N],B[N],C[N]; void fft(cp*a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){cp w0(cos(Pi/i),k*sin(Pi/i));for(int j=0;j<len;j+=(i<<1)){cp w(1,0);for(int o=0;o<i;o++,w=w*w0){cp w1 = a[j+o],w2 = a[j+o+i]*w;a[j+o] = w1+w2;a[j+o+i] = w1-w2;}}}if(k==-1)for(int i=0;i<len;i++)a[i].x/=len; } cp a[N],b[N],c[N],d[N],e[N],f[N],g[N],h[N]; void mtt() {int M = 32768;for(int i=0;i<max(n,m);i++){a[i].x = A[i]/M,b[i].x = A[i]%M;c[i].x = B[i]/M,d[i].x = B[i]%M;}fft(a,lim,1),fft(b,lim,1),fft(c,lim,1),fft(d,lim,1);for(int i=0;i<lim;i++){e[i] = a[i]*c[i],f[i] = a[i]*d[i];g[i] = b[i]*c[i],h[i] = b[i]*d[i];}fft(e,lim,-1),fft(f,lim,-1),fft(g,lim,-1),fft(h,lim,-1);for(int i=0;i<lim;i++)C[i] = (((ll)(e[i].x+0.1))%MOD*M%MOD*M%MOD+((ll)(f[i].x+0.1))%MOD*M%MOD+((ll)(g[i].x+0.1))%MOD*M%MOD+((ll)(h[i].x+0.1))%MOD)%MOD; } int main() {read(n),read(m),read(MOD);n++,m++;init();for(int i=0;i<n;i++)read(A[i]);for(int i=0;i<m;i++)read(B[i]);mtt();for(int i=0;i<n+m-1;i++)printf("%lld ",C[i]);puts("");return 0; }
时间复杂度$O(nlogn)$。大常数。
(4)更强的$MTT$
参见myy论文。
void mtt(int*F,int*G,int*H,int len) {get_lim(len<<1);for(register int i=0;i<lim;++i)a[i]=b[i]=cp(0,0);for(register int i=0;i<len;++i)a[i]=cp(F[i]&(M-1),F[i]>>15),b[i]=cp(G[i]&(M-1),G[i]>>15);fft(a,lim,1),fft(b,lim,1);for(register int i=0;i<lim;++i){int j = (lim-i)&(lim-1);c[j]=cp(0.5*(a[i].x+a[j].x),0.5*(a[i].y-a[j].y))*b[i];d[j]=cp(0.5*(a[i].y+a[j].y),0.5*(a[j].x-a[i].x))*b[i];}fft(c,lim,1),fft(d,lim,1);for(register int i=0;i<lim;++i){int kaa = (ll)(c[i].x/lim+0.5)%MOD;int kab = (ll)(c[i].y/lim+0.5)%MOD;int kba = (ll)(d[i].x/lim+0.5)%MOD;int kbb = (ll)(d[i].y/lim+0.5)%MOD;Mod(H[i] = ((ll)kaa+((ll)(kab+kba)<<15)%MOD+((ll)kbb<<30)%MOD)%MOD+MOD);} }
4.多项式求导、积分
不会求导积分建议出门左转高中数学教材。
void down(ll*a,int len) {for(int i=0;i+1<len;i++)a[i]=a[i+1]*(i+1)%MOD;a[len-1]=0; } void up(ll*a,int len) {for(int i=len-1;i>0;i--)a[i]=a[i-1]*inv(i)%MOD;a[0] = 0; }
求导时间复杂度$O(n)$,积分$O(n)$或者$O(nlog)$。
5.多项式求逆
洛谷传送门。
洛谷加强传送门。
已知多项式$F(x)$,求$G(x)$满足$$F(x)*G(x)≡1\;(mod\;x^n)$$
一个多项式对应的逆是一个唯一的无穷极的多项式,我们要求的是它的前$n$项。
一般都是倍增法求解。
我们现在知道$$F(x)*G0(x)≡1\;(mod\;x^{\frac{n}{2}})$$
移项:$$F(x)*G0(x)-1≡0\;(mod\;x^{\frac{n}{2}})$$
搞模的次数,整体平方:$$(F(x)*G0(x)-1)^2≡0\;(mod\;x^n)$$
打开:$$F^2*G0^2-2*F*G0+1≡0\;(mod\;x^n)$$
两边乘上$G$,得$$F*G0^2-2*G0+G≡0\;(mod\;x^n)$$
再移项:$$G≡2*G0-F*G0^2$$
这个式子减号左边是$\frac{n}{2}$次的,减号右边是$n$次的。
注意右边乘的话先$G*G$再乘$F$,注意长度。
void get_inv(int len,int*F,int*G) {if(len==1){G[0]=fastpow(F[0],MOD-2);return ;}get_inv(len>>1,F,G),mtt(G,G,T,len>>1),mtt(T,F,H,len);for(register int i=0;i<len;++i)Mod(G[i]=G[i]*2%MOD-H[i]+MOD); }
边界条件$G_0= \frac{1}{F_0}$,时间复杂度$O(nlogn)$。
6.多项式整除、取余
洛谷传送门。
给出$F(x),G(x)$,其中$F(x)$是$n$次的,$G(x)$是$m$次的。
求$$F(x)=Q(x)*G(x)+R(x)$$
$R(x)$次数$m-1$,$Q(x)$次数$n-m$。
神奇变换:$$F(\frac{1}{x})=Q(\frac{1}{x})*G(\frac{1}{x})+R(\frac{1}{x})$$
然后同乘$x^n$,发现$$F(\frac{1}{x})*x^n=Q(\frac{1}{x})*x^{n-m}*G(\frac{1}{x})*x^{m}+R(\frac{1}{x})*x^n$$
$$F_R(x)=Q_R(x)*G_R(x)+R_R(x)$$
其中$F_R(x)$表示将$F$系数反转得到的多项式,$Q_R(x)$、$G_R(x)$同理。
但是我们发现$R(x)$是$m-1$次的,乘完之后后面会有一堆0。所以$$F_R(x)≡Q_R(x)*G_R(x)\;(mod\;x^{n-m})$$
反过来求逆就好了。
把$Q(x)$求出来之后求$R(x)$就是多项式乘法、多项式减法。
时间复杂度$O(nlogn)$。
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; typedef long long ll; const int N = 600050; const int MOD = 998244353; template<typename T> inline void read(T&x) {T f = 1,c = 0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}x = f*c; } ll fastpow(ll x,int y) {ll ret = 1;while(y){if(y&1)ret=ret*x%MOD;x=x*x%MOD;y>>=1;}return ret; } int n,m,to[N],lim,L; void ntt(ll*a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){ll w0 = fastpow(3,(MOD-1)/(i<<1));for(int j=0;j<len;j+=(i<<1)){ll w = 1;for(int o=0;o<i;o++,w=w*w0%MOD){ll w1 = a[j+o],w2 = a[j+o+i]*w%MOD;a[j+o] = (w1+w2)%MOD;a[j+o+i] = (w1-w2+MOD)%MOD;}}}if(k==-1){for(int i=1;i<len>>1;i++)swap(a[i],a[len-i]);ll inv = fastpow(len,MOD-2);for(int i=0;i<len;i++)a[i]=a[i]*inv%MOD;} } ll f[N],g[N],a[N],b[N],c[N]; void sol(int len) {if(len==1){g[0] = fastpow(f[0],MOD-2);return ;}int mid = (len+1)>>1;sol(mid);lim = 1,L = 0;while(lim<=2*len)lim<<=1,L++;for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1)));for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=f[i];for(int i=0;i<mid;i++)b[i]=g[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<len;i++)g[i]=(2*g[i]%MOD-c[i]+MOD)%MOD; } void mul(ll*A,ll*B,int len) {lim = 1,L = 0;while(lim<=2*len)lim<<=1,L++;for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1)));for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=A[i],b[i]=B[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD;ntt(c,lim,-1); } ll F[N],G[N],H[N],R[N],F0[N],G0[N]; int main() {read(n),read(m),n++,m++;for(int i=0;i<n;i++)read(F[i]);for(int i=0;i<m;i++)read(G[i]);memcpy(F0,F,sizeof(F0));memcpy(G0,G,sizeof(G0));reverse(F,F+n);reverse(G,G+m);for(int i=0;i<m;i++)f[i]=G[i];sol(n-m+1);mul(F,g,n-m+1);for(int i=0;i<n-m+1;i++)H[i]=c[i];reverse(H,H+n-m+1);mul(H,G0,n);for(int i=0;i<n-m+1;i++)printf("%lld ",H[i]);puts("");for(int i=0;i<m-1;i++)printf("%lld ",(F0[i]-c[i]+MOD)%MOD);puts("");return 0; }
7.多项式开根
洛谷传送门。
这个黑科技是小朋友与二叉树之后大佬们搞出来的。
已知$F(x)$,求$G(x)$满足$$G(x)*G(x)≡F(x)\;(mod\;x^n)$$
和多项式的逆一样,一个多项式的根是惟一的。
依然考虑倍增求解。
假设我们已知$$H(x)*H(x)≡F(x)\;(mod\;x^{\frac{n}{2}})$$,要求$G(x)$。
常规移项平方:$$(H^2(x)-F(x))^2≡0\;(mod\;x^n)$$
然后是一个神奇操作:$$(H^2(x)+F(x))^2≡4F(x)H^2(x)\;(mod\;x^n)$$
然后再移项:$$\frac{(H^2(x)+F(x))^2}{4H^2(x)} ≡ F(x) \; (mod\;x^n)$$
然后惊奇的发现左边是平方形式,所以$$G(x) ≡ \frac{H^2(x)+F(x)}{2H(x)} \; (mod\;x^n)$$
递归的时候套一个多项式求逆即可(反正才四行),注意清数组不然可能会死。
时间复杂度$T(n)=T(n/2)+O(nlogn)=O(nlogn)$,大常数。
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; typedef long long ll; const int N = 600050; const int MOD = 998244353; template<typename T> inline void read(T&x) {T f = 1,c = 0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}x = f*c; } ll fastpow(ll x,int y) {ll ret = 1;while(y){if(y&1)ret=ret*x%MOD;x=x*x%MOD;y>>=1;}return ret; } int to[N],lim,L; void ntt(ll*a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){ll w0 = fastpow(3,(MOD-1)/(i<<1));for(int j=0;j<len;j+=(i<<1)){ll w = 1;for(int o=0;o<i;o++,w=w*w0%MOD){ll w1 = a[j+o],w2 = a[j+o+i]*w%MOD;a[j+o] = (w1+w2)%MOD;a[j+o+i] = (w1-w2+MOD)%MOD;}}}if(k==-1){for(int i=1;i<len>>1;i++)swap(a[i],a[len-i]);ll inv = fastpow(len,MOD-2);for(int i=0;i<len;i++)a[i]=a[i]*inv%MOD;} } int n; ll a[N],b[N],c[N]; void get_lim(int len) {lim = 1,L = 0;while(lim<=len)lim<<=1,L++;for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1))); } void mul(ll*A,int Alen,ll*B,int Blen) {get_lim(Alen+Blen);for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<Alen;i++)a[i]=A[i];for(int i=0;i<Blen;i++)b[i]=B[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD;ntt(c,lim,-1); } ll F[N],G[N],H[N]; void get_inv(int len) {if(len==1){H[0] = fastpow(G[0],MOD-2);return ;}int mid = (len+1)>>1;get_inv(mid);get_lim(len<<1);for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=G[i];for(int i=0;i<mid;i++)b[i]=H[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<len;i++)H[i]=(2*H[i]%MOD-c[i]+MOD)%MOD; } void get_sqrt(int len) {if(len==1){G[0] = 1;return ;}int mid = (len+1)>>1;get_sqrt(mid);for(int i=0;i<len;i++)H[i]=0;get_inv(len);mul(G,mid,G,mid);for(int i=0;i<len;i++)G[i]=(c[i]+F[i])%MOD;mul(G,len,H,len);for(int i=0;i<len;i++)G[i]=c[i]*((MOD+1)/2)%MOD; } int main() {read(n);for(int i=0;i<n;i++)read(F[i]);get_sqrt(n);for(int i=0;i<n;i++)printf("%lld ",G[i]);puts("");return 0; }
8.多项式对数函数(ln)
洛谷传送门。
已知$F(x)$,求$$G(x)≡ln(F(x))\;(mod\;x^n)$$
直接求导即可:$$G'(x)≡F'(x)*\frac{1}{F(x)}$$
所以说求导求逆积分即可。
时间复杂度$O(nlogn)$。
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; typedef long long ll; const int N = 600050; const int MOD = 998244353; template<typename T> inline void read(T&x) {T f = 1,c = 0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}x = f*c; } ll fastpow(ll x,int y) {ll ret = 1;while(y){if(y&1)ret=ret*x%MOD;x=x*x%MOD;y>>=1;}return ret; } ll inv(ll x){return fastpow(x,MOD-2);} int to[N],lim,L; void down(ll*a,int len) {for(int i=0;i+1<len;i++)a[i]=a[i+1]*(i+1)%MOD;a[len-1]=0; } void up(ll*a,int len) {for(int i=len-1;i>0;i--)a[i]=a[i-1]*inv(i)%MOD;a[0] = 0; } void ntt(ll*a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){ll w0 = fastpow(3,(MOD-1)/(i<<1));for(int j=0;j<len;j+=(i<<1)){ll w = 1;for(int o=0;o<i;o++,w=w*w0%MOD){ll w1 = a[j+o],w2 = a[j+o+i]*w%MOD;a[j+o] = (w1+w2)%MOD;a[j+o+i] = (w1-w2+MOD)%MOD;}}}if(k==-1){for(int i=1;i<len>>1;i++)swap(a[i],a[len-i]);ll inv = fastpow(len,MOD-2);for(int i=0;i<len;i++)a[i]=a[i]*inv%MOD;} } int n; ll a[N],b[N],c[N]; ll F[N],G[N]; void get_inv(int len) {if(len==1){G[0] = inv(F[0]);return ;}int mid = (len+1)>>1;get_inv(mid);lim = 1,L = 0;while(lim<2*len)lim<<=1,L++;for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1)));for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=F[i];for(int i=0;i<mid;i++)b[i]=G[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<len;i++)G[i]=(2*G[i]%MOD-c[i]+MOD)%MOD; } int main() {read(n);for(int i=0;i<n;i++)read(F[i]);get_inv(n);down(F,n);lim = 1,L = 0;while(lim<=2*n)lim<<=1,L++;for(int i=1;i<lim;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1)));for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<n;i++)a[i]=F[i];for(int i=0;i<n;i++)b[i]=G[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<n;i++)F[i]=c[i];up(F,n);for(int i=0;i<n;i++)printf("%lld ",F[i]);puts("");return 0; }
9.多项式指数函数(exp)
洛谷传送门。
已知$F(x)$,求$$G(x)≡e^{F(x)}\;(mod\;x^n)$$
附加牛顿迭代:$x=x'-\frac{f(x')}{f(x')'}$
牛迭原理?一张图:
回到问题,依然倍增求解,假设已知$$H(x)≡e^{F(x)}\;(mod\;x^{\frac{n}{2}})$$
先对式子取ln再移项:$$ln(H(x))-F(x)≡0\;(mod\;x^{\frac{n}{2}})$$
然后呢……
我们把它当做$ln(x)-k=0$,问题转化为式子求零点,那么把牛迭套进去有这样一个式子:$$G(x)≡H(x)-\frac{ln(H(x))-F(x)}{\frac{1}{H(x)}}\;(mod\;x^n)$$
分数太难看了,变一下:$$G(x)≡H(x)*(1-ln(H(x))+F(x))\;(mod\;x^n)$$
左转把多项式ln板子带过来即可。
复杂度?$T(n)=T(n/2)+O(nlogn)=O(nlogn)$?
常数?卡死人。
我的$ntt$跑$1e6$的数据:
我的$exp$跑$1e5$的数据:
卡常好啊
// luogu-judger-enable-o2 #include<cstdio> #include<cstring> #include<algorithm> using namespace std; typedef unsigned long long ll; const int N = 600050; const int MOD = 998244353; template<typename T> inline void read(T&x) {T f = 1,c = 0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}x = f*c; } ll fastpow(ll x,int y) {ll ret = 1;while(y){if(y&1)ret=ret*x%MOD;x=x*x%MOD;y>>=1;}return ret; } int to[N],lim,L; ll W[N],Inv; void Mod(ll&x){if(x>=MOD)x-=MOD;} void ntt(ll*a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){ll w0 = W[i];for(int j=0;j<len;j+=(i<<1)){ll w = 1;for(int o=0;o<i;o++,w=w*w0%MOD){ll w1 = a[j+o],w2 = a[j+o+i]*w%MOD;Mod(a[j+o] = w1+w2);Mod(a[j+o+i] = w1-w2+MOD);}}}if(k==-1){for(int i=1;i<len>>1;i++)swap(a[i],a[len-i]);for(int i=0;i<len;i++)a[i]=a[i]*Inv%MOD;} } ll a[N],b[N],c[N]; void get_lim(int len) {lim = 1,L = 0;while(lim<=len)lim<<=1,L++;for(int i=1;i<lim;i++)to[i]=((to[i>>1])>>1|((i&1)<<(L-1)));for(int i=1;i<lim;i<<=1)W[i]=fastpow(3,(MOD-1)/(i<<1));Inv = fastpow(lim,MOD-2); } int n; ll A[N],F[N],G[N],H[N];//F,lnF,1/G void get_inv(int len) {if(len==1){H[0] = fastpow(G[0],MOD-2);return ;}int mid = (len+1)>>1;get_inv(mid);get_lim(len<<1);for(int i=0;i<=lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=G[i];for(int i=0;i<mid;i++)b[i]=H[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<len;i++)Mod(H[i]=2*H[i]%MOD-c[i]+MOD); } void down(ll*a,int len) {for(int i=0;i<len;i++)a[i]=a[i+1]*(i+1)%MOD;a[len-1]=0; } void up(ll*a,int len) {for(int i=len-1;i;i--)a[i]=a[i-1]*fastpow(i,MOD-2)%MOD;a[0] = 0; } void get_ln(int len) {for(int i=0;i<len;i++)G[i]=F[i],H[i]=0;get_inv(len);down(G,len);get_lim(len<<1);for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=G[i],b[i]=H[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<len;i++)G[i]=c[i];up(G,len); } void get_exp(int len) {if(len==1){F[0] = 1;return ;}int mid = (len+1)>>1;get_exp(mid);for(int i=0;i<len;i++)G[i]=0;get_ln(len);get_lim(len<<1);for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=F[i],b[i]=(A[i]-G[i]+MOD)%MOD;Mod(++b[0]);ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=a[i]*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<len;i++)F[i]=c[i]; } int main() {read(n);for(int i=0;i<n;i++)read(A[i]);get_exp(n);for(int i=0;i<n;i++)printf("%llu ",F[i]);puts("");return 0; }
10.多项式快速幂
洛谷传送门。
已知$A(x)$,求$B(x)≡A^k(x)\;(mod\;x^n)$。
直接取$ln$,乘完再$exp$回去……
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; typedef long long ll; const int N = 500050; const int MOD = 998244353; template<typename T> inline void read(T&x) {T f = 1,c = 0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){c=c*10+ch-'0';ch=getchar();}x = f*c; } template<typename T> inline void Mod(T&x){if(x>=MOD)x-=MOD;} int modread() {int ret = 0;char ch=getchar();while(ch<'0'||ch>'9'){ch=getchar();}while(ch>='0'&&ch<='9'){Mod(ret=10ll*ret%MOD+ch-'0');ch=getchar();}return ret; } int fastpow(int x,int y) {int ret = 1;while(y){if(y&1)ret=1ll*ret*x%MOD;x=1ll*x*x%MOD;y>>=1;}return ret; } int inv(int x){return fastpow(x,MOD-2);} int n,k,to[N],lim,L,LL[N],ny[N],jc[N],jny[N]; int init(int n) {lim = LL[2] = 1;while(lim<=n)lim<<=1,LL[lim<<1]=LL[lim]+1;ny[1] = jc[0] = jc[1] = jny[0] = jny[1] = 1;for(int i=2;i<=lim;i++){ny[i] = 1ll*(MOD-MOD/i)*ny[MOD%i]%MOD;jc[i] = 1ll*jc[i-1]*i%MOD;jny[i] = 1ll*jny[i-1]*ny[i]%MOD;}return lim; } void get_lim(int len) {lim = len,L = LL[len];for(int i=1;i<len;i++)to[i]=((to[i>>1]>>1)|((i&1)<<(L-1))); } void ntt(int*a,int len,int k) {for(int i=0;i<len;i++)if(i<to[i])swap(a[i],a[to[i]]);for(int i=1;i<len;i<<=1){int w0 = fastpow(3,(MOD-1)/(i<<1));for(int j=0;j<len;j+=(i<<1)){int w = 1;for(int o=0;o<i;o++,w=1ll*w*w0%MOD){int w1 = a[j+o],w2 = 1ll*a[j+o+i]*w%MOD;Mod(a[j+o] = w1+w2);Mod(a[j+o+i] = w1+MOD-w2);}}}if(k==-1){for(int i=1;i<len>>1;i++)swap(a[i],a[len-i]);int Inv = inv(len);for(int i=0;i<len;i++)a[i]=1ll*a[i]*Inv%MOD;} } int a[N],b[N],c[N],I[N],T[N],Ln[N]; void mul(int*F,int*G,int len) {get_lim(len<<1);for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=F[i],b[i]=G[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=1ll*a[i]*b[i]%MOD;ntt(c,lim,-1); } void get_d(int*F,int*G,int len) {for(int i=1;i<len;i++)G[i-1]=1ll*F[i]*i%MOD;G[len-1]=0; } void get_j(int*F,int*G,int len) {for(int i=1;i<len;i++)G[i]=1ll*F[i-1]*ny[i]%MOD;G[0] = 0; } void get_inv(int*F,int*G,int len) {if(len==1){G[0]=inv(F[0]);return ;}get_inv(F,G,len>>1);get_lim(len<<1);for(int i=0;i<lim;i++)a[i]=b[i]=0;for(int i=0;i<len;i++)a[i]=F[i];for(int i=0;i<len>>1;i++)b[i]=G[i];ntt(a,lim,1),ntt(b,lim,1);for(int i=0;i<lim;i++)c[i]=1ll*a[i]*b[i]%MOD*b[i]%MOD;ntt(c,lim,-1);for(int i=0;i<len;i++)G[i]=(2ll*G[i]%MOD+MOD-c[i])%MOD; } void get_ln(int*F,int*G,int len) {for(int i=0;i<len;i++)I[i]=0;get_inv(F,I,len),get_d(F,T,len);mul(I,T,len);for(int i=0;i<len;i++)T[i]=c[i];get_j(T,G,len); } void get_exp(int*F,int*G,int len) {if(len==1){G[0]=1;return ;}get_exp(F,G,len>>1);get_ln(G,Ln,len);for(int i=0;i<len;i++)Mod(Ln[i]=F[i]+MOD-Ln[i]);Mod(++Ln[0]);mul(Ln,G,len);for(int i=0;i<len;i++)G[i]=c[i]; } void fastpow(int*F,int*G,int len,int k) {get_ln(F,F,len);for(int i=0;i<len;i++)F[i]=1ll*F[i]*k%MOD;get_exp(F,G,len); } int F[N],G[N]; int main() { // freopen("tt.in","r",stdin);read(n),k=modread();for(int i=0;i<n;i++)read(F[i]);int mx = init(n);fastpow(F,G,mx,k);for(int i=0;i<n;i++)printf("%d ",G[i]);puts("");return 0; }
大概就这些了吧。
(是时候总结一下常用卡常技巧了)