[atARC124F]Chance Meeting

为了方便,不妨先将$n$和$m$都减小1,其意义即为移动的次数

注意到老鼠向下移动和猫向上移动对于第2个条件是等价的,对于第1个条件即要求都恰好移动$n$次,那么对应的方案数即为${2n\choose n}$,乘上此系数后不妨将两种操作都看作仅有老鼠向下移动$2n$次

此时,即猫只能向右移动,因此相遇的位置必然在第$n+1$行

定义$f(x)$表示双方最终位于$(n+1,x+1)$且不存在一次操作后双方在同一个位置上的方案数,枚举双方相遇的位置,不难得到答案即为${2n\choose n}\sum_{i=0}^{m}f(i)f(m-i)$

先不考虑不在同一个位置上的条件,总方案数即为${n+2x\choose x}{n+x\choose n}$,将其记作$g(x)$,对于不合法的方案不妨去枚举其上一次相遇的位置,此时限制即之后不再相遇

具体的,可以看作在坐标系中从$(0,0)$走到$(n,0)$,每一步可以从$(x,y)$移动到$(x+1,y\pm 1)$,求除了起点和终点以外不与$x$轴有公共点的路径数

不难发现这即为$2H_{n-1}$(其中$H$为卡特兰数),也即有$f(x)=g(x)-2\sum_{i=0}^{x-1}H_{x-i-1}g(i)$

时间复杂度为$o(n+m\log m)$,可以通过

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define N 600005
 4 #define L (1<<19)
 5 #define mod 998244353
 6 #define ll long long
 7 int n,m,ans,fac[N],inv[N],rev[L],a[L],b[L],g[N],f[N];
 8 int C(int n,int m){
 9     return (ll)fac[n]*inv[m]%mod*inv[n-m]%mod;
10 }
11 int qpow(int n,int m){
12     int s=n,ans=1;
13     while (m){
14         if (m&1)ans=(ll)ans*s%mod;
15         s=(ll)s*s%mod;
16         m>>=1;
17     }
18     return ans;
19 }
20 void ntt(int *a,int p){
21     for(int i=0;i<L;i++)
22         if (i<rev[i])swap(a[i],a[rev[i]]);
23     for(int i=2;i<=L;i<<=1){
24         int s=qpow(3,(mod-1)/i);
25         if (p)s=qpow(s,mod-2);
26         for(int j=0;j<L;j+=i)
27             for(int k=0,ss=1;k<(i>>1);k++,ss=(ll)ss*s%mod){
28                 int x=a[j+k],y=(ll)a[j+k+(i>>1)]*ss%mod;
29                 a[j+k]=(x+y)%mod,a[j+k+(i>>1)]=(x-y+mod)%mod;
30             }
31     }
32     if (p){
33         int s=qpow(L,mod-2);
34         for(int i=0;i<L;i++)a[i]=(ll)a[i]*s%mod;
35     }
36 }
37 int main(){
38     scanf("%d%d",&n,&m);
39     n--,m--;
40     fac[0]=inv[0]=inv[1]=1;
41     for(int i=1;i<N;i++)fac[i]=(ll)fac[i-1]*i%mod;
42     for(int i=2;i<N;i++)inv[i]=(ll)(mod-mod/i)*inv[mod%i]%mod;
43     for(int i=1;i<N;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
44     for(int i=0;i<L;i++)rev[i]=(rev[i>>1]>>1)+((i&1)*(L>>1));
45     for(int i=0;i<=m;i++)a[i]=g[i]=(ll)C(n+(i<<1),i)*C(n+i,n)%mod;
46     for(int i=0;i<m;i++)b[i+1]=(ll)C((i<<1),i)*fac[i]%mod*inv[i+1]%mod;
47     ntt(a,0),ntt(b,0);
48     for(int i=0;i<L;i++)a[i]=(ll)a[i]*b[i]%mod;
49     ntt(a,1);
50     for(int i=0;i<=m;i++)f[i]=(g[i]-2*a[i]%mod+mod)%mod;
51     for(int i=0;i<=m;i++)ans=(ans+(ll)f[i]*f[m-i])%mod;
52     ans=(ll)ans*C((n<<1),n)%mod;
53     printf("%d\n",ans);
54     return 0;
55 } 
View Code

 

posted @ 2021-09-11 08:35  PYWBKTDA  阅读(71)  评论(0编辑  收藏  举报