ABC240G Teleporting Takahashi[组合数学]

题意：走$N$步从$(0,0,0)$走到$(X,Y,Z)$的不同路径条数（每一步不能不走且步长为1）
不失一般性的我们可以设$(X,Y,Z>0)$
首先考虑走$(X+Y+2k)$步从$(0,0)$走到$(X,Y)$
枚举$X$正方向的步数$(X+i)$，总的走法就是
$$C_{x+y+2k}^k\sum _{i=0}^k{C}_k^i\times C_{x+y+k}^{x+i}=C_{x+y+2k}^k\sum _{i=0}^k{C}_k^{k-i}\times C_{x+y+k}^{x+i}=C_{x+y+2k}^k\times C_{x+y+2k}^{y+k}$$
其中$C_{x+y+2k}^k$枚举了多走的$k$步的位置，$C_k^i\times C_{x+y+k}^{x+i}$分别枚举了$X$方向回退的$i$步的位置和$X$方向走的$(X+i)$步位置
其中等式${\sum }_{i=0}^k{C}_k^{k-i}\times C_{x+y+k}^{x+i}=C_{x+y+2k}^{y+k}$是著名的范德蒙恒等式
一般形式为${\sum }_{i=0}^k{C}_m^{k-i}\times C_n^i=C_{m+n}^k$
推导过程如下：

考虑在一堆和为$(m+n)$数量的物品中选$k$个的方案数，考虑前$n$个中选了$i$个，后$m$个中选了$(k-i)$个的方案，枚举所有$i$求和就得到了答案。

接下来考虑$Z$方向上的情况，剩余未考虑的步数为$N-(X+Y+2k)$，这些步数中朝$Z$负方向的有$(N-(X+Y+Z+2k))/2$步，其余为正方向
则对于某个$k$，方案为
$$C_{x+y+2k}^k\times C_{x+y+2k}^{y+k}\times C_N^{N-(x+y+2k)}\times C_{N-(X+Y+2k)}^{(N-(X+Y+Z+2k))/2}$$
枚举所有$k$就得到了答案，复杂度$O(N)$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod=998244353;
ll poww(ll a,ll b)
{
	ll t=1;
	while(b)
	{
		if(b&1)t=a*t%mod;
		a=a*a%mod;
		b>>=1;
	}
	return t;
}
ll inv(ll x)
{
	return poww(x,mod-2);
}
ll P1[10000005],P2[10000005];
void init()
{
	P1[0]=1;
	for(int i=1;i<=10000000;i++)P1[i]=1LL*P1[i-1]*i%mod;
	P2[10000000]=inv(P1[10000000]);
	for(int i=9999999;i>=0;i--)P2[i]=1LL*P2[i+1]*(i+1)%mod;
}
ll C(ll n,ll m)
{
	return 1LL*P1[n]*P2[m]%mod*P2[n-m]%mod;
}
int main()
{
	init();
	ll N,X,Y,Z;
	scanf("%lld%lld%lld%lld",&N,&X,&Y,&Z);
	X=abs(X);
	Y=abs(Y);
	Z=abs(Z);
	ll left=N-X-Y-Z;
	if(left<0||(left%2!=0))printf("0");
	else
	{
		ll ans=0;
		for(int k=0;k<=left/2;k++)
		{
			ans=(ans+1LL*C(X+Y+2*k,k)*
				C(X+Y+2*k,Y+k)%mod*
				C(N,N-(X+Y+2*k))%mod*
				C(N-(X+Y+2*k),(N-(X+Y+2*k+Z))/2)%mod
				)%mod;
		}
		printf("%lld\n",ans);
	}	
    return  0;
}