有三堆卡牌,牌数分别为N,M,K
每张牌有一个字母(’a’、’b’、’c’)表示下一个拿哪一堆。
现在要求第一堆首先拿完。求方案数。
首先,这道题有很多角度可以入手。但最简单的方法是,根据拿的牌的类型计算。
如果第一堆拿完,则拿的顺序中必然有n个a,且b的数量比m小,c的数量比k小。
但是最后一个必须限定为a,所以这部分拿了的方案数应为 Cn−1n−1+i C n − 1 + i n − 1 (i表示多拿的b和c的总和)
然后还要从这i个中选哪些是b,哪些是c,所以还要乘上 ∑0≤x<m,0≤y<k[x+y=i]Cxi ∑ 0 ≤ x < m , 0 ≤ y < k [ x + y = i ] C i x
所以有如下式子:
Ans=∑i<m+ki=0Cn−1n−1+i3m+k−i∑0≤x<m,0≤y<k[x+y=i]Cxi A n s = ∑ i = 0 i < m + k C n − 1 + i n − 1 3 m + k − i ∑ 0 ≤ x < m , 0 ≤ y < k [ x + y = i ] C i x
然后这就可以拿所谓的部分分了。
接下来一般来说都会想怎么改进式子,然而。。。这道题的方法是,直接枚举前半部分(即i的大小),想办法快速转移后半部分。
后半部分可以分三种情况讨论(设 k≤m k ≤ m ):
注:下面x表示的是拿c的个数
1、 ∑x≤ix=0Cxi(i<k) ∑ x = 0 x ≤ i C i x ( i < k ) 即c的数量比i还多,所以任意拿就可以了
2、 ∑x<kx=0Cxi(k≤i<m) ∑ x = 0 x < k C i x ( k ≤ i < m ) 即c的数量不超过i,这时c就不能拿超过k个
3、 ∑x<kx=i−m+1Cxi(m≤i<m+k) ∑ x = i − m + 1 x < k C i x ( m ≤ i < m + k ) 即c的数量不超过i,b的数量也不超过i,这时c不能拿大于等于k个,b不能拿大于等于m个。
考虑优化转移。在第一种情况下,i变为i+1,只需要把总和乘2即可。
在第二种情况下,i变为i+1,把总和乘以2再减去 Cki C i k
在第三种情况下,i变为i+1,把总和乘以2再减去 Cki+Cmi C i k + C i m
用杨辉三角的性质来解释就很容易了。
#include
#include
#include
#include
#define SF scanf
#define PF printf
#define MAXN 900010
#define MOD 1000000007
using namespace std;
typedef long long ll;
ll inv[MAXN],fac[MAXN],pow3[MAXN];
ll fsp(ll x,int y){
ll res=1;
while(y){
if(y&1)
res=res*x%MOD;
x=x*x%MOD;
y>>=1;
}
return res;
}
ll C(int x,int y){
return fac[x]*inv[y]%MOD*inv[x-y]%MOD;
}
int main(){
int n,m,k;
SF("%d%d%d",&n,&m,&k);
pow3[0]=1;
fac[0]=1;
for(int i=1;i<=n+m+k;i++){
pow3[i]=pow3[i-1]*3ll%MOD;
fac[i]=fac[i-1]*i%MOD;
}
inv[n+m+k]=fsp(fac[n+m+k],MOD-2);
for(int i=n+m+k;i>0;i--)
inv[i-1]=inv[i]*i%MOD;
n--;
if(mm,k);
ll j=1,ans=0;
for(int i=0;i<=m+k;i++){
ans=(ans+C(n+i,n)*pow3[m+k-i]%MOD*j)%MOD;
if(i*2ll%MOD;
else if(i<m)
j=(j*2ll-C(i,k))%MOD;
else
j=(j*2ll-C(i,k)-C(i,m))%MOD;
}
ans=(ans+MOD)%MOD;
PF("%lld",ans);
}