洛谷 P4512 [模板] 多项式除法

题目：https://www.luogu.org/problemnew/show/P4512

看博客：https://www.cnblogs.com/owenyu/p/6724611.html

https://www.cnblogs.com/Mychael/p/9216906.html

注意取模那里的 NTT 范围就是模数的次数；

各处注意一下对系数数组取模(超出的位置赋0)。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=(<<),g=,mod=;
int n,m,a[xn],b[xn],d[xn],r[xn],rev[xn],c[xn],t[xn];
int rd()
{
  int ret=,f=; char ch=getchar();
  while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
  while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
  return f?ret:-ret;
}
ll pw(ll a,int b)
{
  ll ret=;
  for(;b;b>>=,a=(a*a)%mod)if(b&)ret=(ret*a)%mod;
  return ret;
}
int upt(int x){while(x>=mod)x-=mod; while(x<)x+=mod; return x;}
void ntt(int *a,int tp,int lim)
{
  for(int i=;i<lim;i++)
    if(i<rev[i])swap(a[i],a[rev[i]]);
  for(int mid=;mid<lim;mid<<=)
    {
      int wn=pw(g,(mod-)/(mid<<));
      if(tp==-)wn=pw(wn,mod-);
      for(int j=,len=(mid<<);j<lim;j+=len)
    {
      int w=;
      for(int k=;k<mid;k++,w=(ll)w*wn%mod)
        {
          int x=a[j+k],y=(ll)w*a[j+mid+k]%mod;
          a[j+k]=upt(x+y); a[j+mid+k]=upt(x-y);
        }
    }
    }
  if(tp==)return; int inv=pw(lim,mod-);
  for(int i=;i<lim;i++)a[i]=(ll)a[i]*inv%mod;
}
void inv(int n,int *a,int *b)
{
  if(n==){b[]=pw(a[],mod-); return;}
  inv((n+)>>,a,b);
  int lim=,l=;
  while(lim<=n+n)lim<<=,l++;
  for(int i=;i<lim;i++)rev[i]=((rev[i>>]>>)|((i&)<<(l-)));
  for(int i=;i<n;i++)c[i]=a[i];
  for(int i=n;i<lim;i++)c[i]=;
  ntt(c,,lim); ntt(b,,lim);
  for(int i=;i<lim;i++)b[i]=((ll)-(ll)c[i]*b[i])%mod*b[i]%mod;
  ntt(b,-,lim);
  for(int i=n;i<lim;i++)b[i]=;
}
void work()
{
  reverse(a,a+n+); reverse(b,b+m+);
  int s=n-m+; inv(s,b,t);
  int lim=,l=;
  while(lim<=s+s)lim<<=,l++;//s
  for(int i=;i<lim;i++)rev[i]=((rev[i>>]>>)|((i&)<<(l-)));
  for(int i=;i<s;i++)c[i]=a[i];
  for(int i=s;i<lim;i++)c[i]=;//
  ntt(c,,lim); ntt(t,,lim);
  for(int i=;i<lim;i++)d[i]=(ll)c[i]*t[i]%mod;
  ntt(d,-,lim);
  for(int i=s;i<lim;i++)d[i]=;//%
  reverse(d,d+s);

  reverse(a,a+n+); reverse(b,b+m+);
  lim=,l=;
  while(lim<=n+n)lim<<=,l++;//n
  for(int i=;i<lim;i++)rev[i]=((rev[i>>]>>)|((i&)<<(l-)));
  for(int i=;i<s;i++)c[i]=d[i];
  for(int i=s;i<lim;i++)c[i]=;
  ntt(b,,lim); ntt(c,,lim);
  for(int i=;i<lim;i++)r[i]=(ll)b[i]*c[i]%mod;
  ntt(r,-,lim);
  for(int i=;i<m;i++)r[i]=((a[i]-r[i])%mod+mod)%mod;
  for(int i=m;i<lim;i++)r[i]=;
}
int main()
{
  n=rd(); m=rd();
  for(int i=;i<=n;i++)a[i]=rd();
  for(int i=;i<=m;i++)b[i]=rd();
  work();
  for(int i=;i<=n-m;i++)printf("%d ",d[i]); puts("");
  for(int i=;i<m;i++)printf("%d ",r[i]); puts("");
  return ;
}