题目:
题意:
分别有 n n n个黑棋和 m m m个白棋,任意排列,在满足任意一段内黑棋与白棋数量的差值为 k k k的方案数是多少
分析:
设
f
i
,
j
,
k
1
,
k
2
f_{i,j,k1,k2}
fi,j,k1,k2表示当前黑棋有
i
i
i个,白棋有
j
j
j个,在后缀里黑棋与白棋的最大差值为
k
1
k_1
k1,白棋与黑棋的最大差值为
k
2
k_2
k2
加一个黑棋或白棋,就在对应的
k
x
+
(
−
)
1
k_x+(-)1
kx+(−)1,但出现负数的时候,显然后面另一种颜色的棋子就会在自己的那个位置才开始计算差值,这样才会是最大的,所以就变成
0
0
0好了
代码:
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define LL long long
#define mo 1000000007
using namespace std;
inline LL read()
{
LL s=0,f=1; char c=getchar();
while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {s=s*10+c-'0';c=getchar();}
return s*f;
}
int f[155][155][25][25];
int main()
{
int n=read(),m=read(),k=read();
f[0][0][0][0]=1;
for(int i=0;i<=n;i++)
for(int j=0;j<=m;j++)
for(int k1=0;k1<=min(i,k);k1++)
for(int k2=0;k2<=min(j,k);k2++)
{
(f[i+1][j][k1+1][max(k2-1,0)]+=f[i][j][k1][k2])%=mo;
(f[i][j+1][max(k1-1,0)][k2+1]+=f[i][j][k1][k2])%=mo;
}
int ans=0;
for(int k1=0;k1<=k;k1++)
for(int k2=0;k2<=k;k2++)
(ans+=f[n][m][k1][k2])%=mo;
cout<<ans;
return 0;
}