洛谷 4721 【模板】分治 FFT——分治FFT / 多项式求逆

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

分治FFT：https://www.cnblogs.com/bztMinamoto/p/9749557.html

　　　　　https://blog.csdn.net/VictoryCzt/article/details/82939586

不知为何自己的总是很慢。

觉得是 n 和 m 表示次数的话，len<=n+m；n 和 m 表示项数的话，len<n+m；应该是这样？

这里是 mid-L+1 项和 R-L+1 项的两个多项式相乘，所以 len < (mid-L+1)+(R-L+1)；

但这样很慢；发现那个 g[0] = 0 每次浪费了一个位置；所以把 g 的标号都减小1，位置 i 对应的位置标号自然也减小了1；微妙地快了一点。

发现用到的最高次项也只是 R-L-1 次；所以 len<= R-L-1 即可！快了一倍。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+,M=N<<,mod=;
int len,r[M],f[M],g[M],a[M],b[M];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
void upd(int &x){x>=mod?x-=mod:;}
int pw(int x,int k)
{int ret=;while(k){if(k&)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=;}return ret;}
void ntt(int *a,bool fx)
{
  for(int i=;i<len;i++)
    if(i<r[i])swap(a[i],a[r[i]]);
  for(int R=;R<=len;R<<=)
    {
      int Wn=pw( ,(mod-)/R );
      if(fx)Wn=pw( Wn,mod- );
      for(int i=,m=R>>;i<len;i+=R)
    for(int j=,w=;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;  upd(a[i+j]);
        a[i+m+j]=x+mod-y;  upd(a[i+m+j]);
      }
    }
  if(!fx)return; int inv=pw( len,mod- );
  for(int i=;i<len;i++)a[i]=(ll)a[i]*inv%mod;
}
void solve(int L,int R)
{
  if(L==R)return;
  int mid=L+R>>;
  solve(L,mid);
  int d=R-L-,i,j;
  for(i=,j=L;j<=mid;i++,j++)a[i]=f[j];// d+=i-1;
  for(i=,j=R-L;i<j;i++)b[i]=g[i+];//  d+=i-1;//+1
  for(len=;len<=d;len<<=);
  for(i=;i<len;i++)r[i]=(r[i>>]>>)+((i&)?len>>:);
  for(i=mid-L+;i<len;i++)a[i]=;  for(i=R-L+;i<len;i++)b[i]=;
  ntt(a,); ntt(b,);
  for(i=;i<len;i++)a[i]=(ll)a[i]*b[i]%mod;
  ntt(a,);
  for(i=mid+,j=i-L-;i<=R;i++,j++)f[i]+=a[j],upd(f[i]);////j=i-L -1
  solve(mid+,R);
}
int main()
{
  int n;n=rdn();for(int i=;i<n;i++)g[i]=rdn();
  f[]=;
  solve(,n-);
  for(int i=;i<n;i++)printf("%d ",f[i]);puts("");
  return ;
}

多项式做法可参见洛谷自带的题解。

F(x) - f[0] = F(x)*G(x)

F(x) = 1/ ( f[0]-G(x) )

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e5+,M=N<<,mod=;
int a[M],b[M],A[M],len,r[M];
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='') ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
void upd(int &x){x>=mod?x-=mod:;}
int pw(int x,int k)
{int ret=;while(k){if(k&)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=;}return ret;}
void ntt(int *a,bool fx)
{
  for(int i=;i<len;i++)
    if(i<r[i])swap(a[i],a[r[i]]);
  for(int R=;R<=len;R<<=)
    {
      int Wn=pw( ,fx?(mod-)-(mod-)/R:(mod-)/R );
      for(int i=,m=R>>;i<len;i+=R)
    for(int j=,w=;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;   upd(a[i+j]);
        a[i+m+j]=x+mod-y;   upd(a[i+m+j]);
      }
    }
  if(!fx)return; int inv=pw(len,mod-);
  for(int i=;i<len;i++)a[i]=(ll)a[i]*inv%mod;
}
void inv(int n)
{
  if(n==){b[]=pw(a[],mod-);return;}
  inv(n+>>);
  for(len=;len<n<<;len<<=);
  for(int i=;i<len;i++)r[i]=(r[i>>]>>)+((i&)?len>>:);
  for(int i=;i<n;i++)A[i]=a[i];  for(int i=n;i<len;i++)A[i]=;
  ntt(A,);  ntt(b,);
  for(int i=;i<len;i++)b[i]=((b[i]<<)-(ll)A[i]*b[i]%mod*b[i])%mod+mod,upd(b[i]);
  ntt(b,);
  for(int i=n;i<len;i++)b[i]=;
}
int main()
{
  int n; n=rdn(); for(int i=;i<n;i++)a[i]=mod-rdn(); a[]=;
  inv(n);
  for(int i=;i<n;i++)printf("%d ",b[i]);puts("");
  return ;
}