[集训队作业2018]uoj 424 count - 组合计数 - 结论

观察到两个序列同构当且仅当建出来的笛卡尔树同构。
一个序列合法当且仅当深度不超过m。
因此问题转为左括号数不超过k的括号序列数量。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<assert.h>
#define mod 998244353
#define lint long long
#define N 2000010
#define A(x,y) (swap(x,y),x--,y++)
#define B(x,y) (swap(x,y),x+=k+1,y-=k+1)
#define P(x) (x<0?x+=mod:0)
#define Q(x) (x>=mod?x-=mod:0)
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
int f[N],fi[N];
inline int fast_pow(int x,int k,int ans=1)
{	for(;k;k>>=1,x=(lint)x*x%mod) (k&1)?ans=(lint)ans*x%mod:0;return ans;	}
inline int prelude(int n)
{
	for(int i=f[0]=1;i<=n;i++) f[i]=(lint)f[i-1]*i%mod;
	fi[n]=fast_pow(f[n],mod-2);
	for(int i=n-1;i>=0;i--) fi[i]=fi[i+1]*(i+1ll)%mod;
	return 0;
}
inline int C(int n,int m)
{
	if(n<0||m<0||n<m) return 0;
	return (lint)f[n]*fi[m]%mod*fi[n-m]%mod;
}
inline int F(int n,int m,int k,int ans=0)//n +1, m -1;
{
	for(int x=n,y=m;x>=0&&y>=0;)
		A(x,y),ans-=C(x+y,x),P(ans),
		B(x,y),ans+=C(x+y,x),Q(ans);
	for(int x=n,y=m;x>=0&&y>=0;)
		B(x,y),ans-=C(x+y,x),P(ans),
		A(x,y),ans+=C(x+y,x),Q(ans);
	return ans+=C(n+m,n),Q(ans),ans;
}
int main()
{
	prelude(N-1);
	int n,m,k;scanf("%d%d",&n,&k),m=n;
	if(k>n) return !printf("0\n");
	if(k<n-m) return !printf("0\n");
	return !printf("%d\n",F(n,m,k));
}