1. 程式人生 > >洛谷 4512 【模板】多項式除法

洛谷 4512 【模板】多項式除法

題目:https://www.luogu.org/problemnew/show/P4512

學習:http://blog.miskcoo.com/2015/05/polynomial-division

getinv裡最好弄一個臨時陣列存 a[ ] ,不要把 a ntt了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+5,M=N<<2
,mod=998244353; int ya[M],yb[M],a[M],b[M],tp[M],bn[M],D[M],R[M],len,r[M]; int rdn() { int ret=0;bool fx=1;char ch=getchar(); while(ch>'9'||ch<'0'){if(ch=='-')fx=0;ch=getchar();} while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=getchar(); return fx?ret:-ret; } void upd(int &x){x>=mod?x-=mod:0
;} int pw(int x,int k) {int ret=1;while(k){if(k&1)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=1;}return ret;} void rev(int n,int *a) { for(int i=0,lm=n+1>>1;i<lm;i++)swap(a[i],a[n-i]); } void ntt(int *a,bool fx) { for(int i=0;i<len;i++) if(i<r[i])swap(a[i],a[r[i]]); for(int R=2
;R<=len;R<<=1) { int Wn=pw(3,(mod-1)/R); if(fx)Wn=pw(Wn,mod-2); for(int i=0,m=R>>1;i<len;i+=R) for(int j=0,w=1;j<m;j++,w=(ll)w*Wn%mod) { int x=a[i+j],y=(ll)w*a[i+m+j]%mod; a[i+j]=x+y; a[i+m+j]=x+mod-y; upd(a[i+j]); upd(a[i+m+j]); } } if(!fx)return; int inv=pw(len,mod-2); for(int i=0;i<len;i++)a[i]=(ll)a[i]*inv%mod; } void gtinv(int n,int *a,int *b) { if(n==1){b[0]=pw(a[0],mod-2);return;} gtinv(n+1>>1,a,b); for(len=1;len<=n<<1;len<<=1); for(int i=0;i<len;i++)r[i]=(r[i>>1]>>1)+((i&1)?len>>1:0); for(int i=0;i<n;i++)tp[i]=a[i]; for(int i=n;i<len;i++)tp[i]=0;/// ntt(tp,0); ntt(b,0); for(int i=0;i<len;i++)b[i]=((b[i]<<1)-(ll)tp[i]*b[i]%mod*b[i])%mod+mod,upd(b[i]); // remember <<1 must add () !!!!!!! ntt(b,1); for(int i=n;i<len;i++)b[i]=0; } int main() { int n,m;n=rdn();m=rdn(); for(int i=0;i<=n;i++)ya[i]=a[i]=rdn(); for(int i=0;i<=m;i++)yb[i]=b[i]=rdn(); rev(n,a); rev(m,b); int d=n-m+1; for(int i=d;i<=n;i++)a[i]=0; for(int i=d;i<=m;i++)b[i]=0; gtinv(d,b,bn); for(len=1;len<=d<<1;len<<=1);//d<<1 not n-m<<1 for(int i=0;i<len;i++)r[i]=(r[i>>1]>>1)+((i&1)?len>>1:0); ntt(a,0); ntt(bn,0); for(int i=0;i<len;i++)D[i]=(ll)a[i]*bn[i]%mod; ntt(D,1); for(int i=d;i<len;i++)D[i]=0; rev(n-m,D); for(len=1;len<=n;len<<=1); for(int i=0;i<len;i++)r[i]=(r[i>>1]>>1)+((i&1)?len>>1:0); for(int i=0,lm=n-m;i<=lm;i++)printf("%d ",D[i]);puts(""); ntt(D,0); ntt(yb,0);//ntt(D) after print_D for(int i=0;i<len;i++)R[i]=(ll)yb[i]*D[i]%mod; ntt(R,1); for(int i=0;i<m;i++)R[i]=ya[i]+mod-R[i],upd(R[i]),printf("%d ",R[i]);puts(""); return 0; }