【牛客挑战赛30D】小A的昆特牌(组合问题抽象到二维平面)

点此看题面

大致题意: 有\(S\)张无编号的牌,可以将任意张牌锻造成\(n\)种步兵或\(m\)种弩兵中的一种,求最后步兵数量大于等于\(l\)小于等于\(r\)的方案数。

暴力式子

首先我们来考虑暴力式子。

假设我们确定了要选\(x\)个步兵数量,然后要求出此时的方案数。

则我们就要使用隔板法

仔细思考,其实我们就相当于要求出把\(x\)个步兵分成\(n\)组把\(S-x\)个步兵分成\(m+1\)组的方案数的乘积。(其中\(m+1\)组指的是\(m\)种弩兵以及不锻造这\(m+1\)种情况)

而这两个要求的东西本质上是一样的。

以把\(x\)个步兵分成\(n\)组为例,考虑到这是无编号的,因此我们完全可以假设每个步兵被分到的组号是递增的。

那也就是说,我们要求把一个长度为\(x\)的序列分割成\(n\)部分(可以为空)的方案数。

而分割成\(n\)部分,就相当于加入了\(n-1\)块隔板。

如果把隔板也看做序列的一部分,则序列总长度就变成了\(x+n-1\),而分割就相当于要在这个序列中选出\(n-1\)个位置。

因此方案数就是:

\[C_{x+n-1}^{n-1}
\]

同理,把\(S-x\)个步兵分成\(m+1\)组的方案数就是:

\[C_{s-x+m}^m
\]

于是它们的乘积就是:

\[C_{x+n-1}^{n-1}*C_{s-x+m}^m
\]

而最终答案就是:

\[\sum_{x=l}^rC_{x+n-1}^{n-1}*C_{s-x+m}^m
\]

抽象问题到二维平面

我们可以发现,这其实就相当于从\((0,0)\)走到\((S,n+m)\),且必须经过点\((l,n)\)下方,不能经过点\((r+1,n)\)下方的方案数。

这其实就相当于用经过点\((l,n)\)下方的方案数减去经过点\((r+1,n)\)下方的方案数。

应该还是比较简单的。

代码

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 10000000
#define X 998244353
#define max(x,y) ((x)>(y)?(x):(y))
#define Inc(x,y) ((x+=(y))>=X&&(x-=X))
using namespace std;
int n,m,s,l,r,Inv[(N<<1)+5],p1[N+5],p2[N+5];
I int C(CI x,CI y)//求组合数
{
if(x<y||x<0) return 0;RI i,res=1;//判断越界
for(i=1;i<=y;++i) res=1LL*res*(x-i+1)%X*Inv[i]%X;//统计答案
return res;//返回答案
}
I int Calc(CI x)//计算经过点(x,n)下方的方案数
{
RI i,res=0;for(p1[0]=i=1;i^n;++i) p1[i]=1LL*p1[i-1]*(x-1+i)%X*Inv[i]%X;//计算p1
for(p2[n-1]=C(s-x+m+1,m+1),i=n-2;~i;--i) p2[i]=1LL*p2[i+1]*(s-x+n+m-i)%X*Inv[n+m-i]%X;//计算p2
for(i=0;i^n;++i) Inc(res,1LL*p1[i]*p2[i]%X);return res;//统计答案
}
int main()
{
RI i,lim;scanf("%d%d%d%d%d",&n,&m,&s,&l,&r);
for(Inv[0]=Inv[1]=1,i=2,lim=max(n,m)<<1;i<=lim;++i) Inv[i]=1LL*(X-X/i)*Inv[X%i]%X;//线性求逆元
return printf("%d",(Calc(l)-Calc(r+1)+X)%X),0;//输出答案
}
上一篇:5.15 牛客挑战赛40 E 小V和gcd树 树链剖分 主席树 树状数组 根号分治


下一篇:NowCoder--牛客练习赛30 C_小K的疑惑