[COGS2287][HZOI 2015]疯狂的机器人（NTT）

最新推荐文章于 2019-02-09 22:13:00 发布

FromATP

最新推荐文章于 2019-02-09 22:13:00 发布

阅读量809

点赞数

分类专栏： BZOJ 总在画柿子的FFT 文章标签： COGS HZOI NTT

本文链接：https://blog.csdn.net/FromATP/article/details/69055731

版权

BZOJ 同时被 2 个专栏收录

112 篇文章 0 订阅

订阅专栏

总在画柿子的FFT

5 篇文章 0 订阅

订阅专栏

=== ===

这里放传送门

=== ===

题解

首先把机器人的路径分成上下和左右两部分考虑。
设上下一共走了i步，方案数为 $c[i]$ 。那么因为最后要回到原点，所以 $i$ 必须是偶数，并且向上和向下分别走了 $i \over 2$ 步。而又因为不能走出第一象限，所以任意时刻向上走的步数不能小于向下走的步数。
可以看出这实际上就是第 $i \over 2$ 项卡特兰数。
那么如果不考虑不走的情况，设一共走了 $i$ 步的方案数为 $f[i]$ ，那么显然 $f[i]=\sum\limits_{j=0}^ic[j]*c[i-j]*C_i^j=c[j]*c[i-j]*{i! \over {j!(i-j)!}}$
设 $g[i]={c[i] \over i!}$ ，那么 $f[i]=i!\sum\limits_{j=1}^ig[j]*g[i-j]$
显然可以用NTT进行加速。
最后统计答案的时候枚举有多少步不走，那么 $ans=\sum\limits_{i=0}^nC_n^i*f[n-i]$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const long long Mod=998244353;
int n,G,L,R[300010],N,M;
long long Ct[300010],mul[300010],g[300010],f[300010],ans,tmp;
long long powww(long long a,int t,long long Mod){
    long long ans=1;a%=Mod;
    while (t!=0){
        if (t&1) ans=(ans*a)%Mod;
        a=(a*a)%Mod;t>>=1;
    }
    return ans;
}
int calc(int x){
    bool flag;
    if (x==2) return 1;
    for (int i=2;;i++){
        flag=true;
        for (int j=2;j*j<x;j++)
          if (powww(i,(x-1)/j,x)==1){
              flag=false;break;
          }
        if (flag==true) return i;
    }
}
long long C(int n,int m){
    long long up,down;
    up=mul[n];down=mul[m]*mul[n-m]%Mod;
    up=up*powww(down,Mod-2,Mod)%Mod;
    return up;
}
void NTT(long long *a,int N,int opt){
    long long wn,w,x,y;
    for (int i=0;i<N;i++)
      if (i<R[i]) swap(a[i],a[R[i]]);
    for (int k=1;k<N;k<<=1){
        wn=powww(G,(Mod-1)/(k<<1),Mod);
        for (int p=(k<<1),i=0;i<N;i+=p){
            w=1;
            for (int j=0;j<k;j++){
                x=a[i+j];y=w*a[i+j+k]%Mod;
                a[i+j]=(x+y)%Mod;a[i+j+k]=(x-y)%Mod;
                w=(w*wn)%Mod;
            }
        }
    }
    if (opt==-1) reverse(a+1,a+N);
}
int main()
{
    scanf("%d",&n);
    G=calc(Mod);mul[0]=1;M=n+n;
    for (N=1;N<=M;N<<=1) L++;
    for (int i=0;i<N;i++)
      R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));
    for (int i=1;i<=N;i++) mul[i]=(mul[i-1]*i)%Mod;
    for (int i=2;i<=n;i+=2)
      Ct[i]=(C(i,i/2)-C(i,i/2-1))%Mod;
    for (int i=0;i<=n;i++)
      g[i]=Ct[i]*powww(mul[i],Mod-2,Mod)%Mod;
    g[0]=1;NTT(g,N,1);
    for (int i=0;i<=N;i++) f[i]=g[i]%Mod*g[i]%Mod;
    NTT(f,N,-1);tmp=powww(N,Mod-2,Mod);
    for (int i=0;i<=N;i++) f[i]=f[i]*tmp%Mod*mul[i]%Mod;
    for (int i=0;i<=n;i++) ans=(ans+f[i]*C(n,i)%Mod)%Mod;
    ans=(ans+Mod)%Mod;
    printf("%I64d\n",ans);
    return 0;

}