题目描述
这次小可可想解决的难题和中国象棋有关,在一个N行M列的棋盘上,让你放若干个炮(可以是0个),使得没有一个炮可以攻击到另一个炮,请问有多少种放置方法。大家肯定很清楚,在中国象棋中炮的行走方式是:一个炮攻击到另一个炮,当且仅当它们在同一行或同一列中,且它们之间恰好 有一个棋子。你也来和小可可一起锻炼一下思维吧!
输入格式
一行包含两个整数N,M,之间由一个空格隔开。
输出格式
总共的方案数,由于该值可能很大,只需给出方案数模9999973的结果。
dp[i][j][k]代表前i行有一个炮的列数为j,两个炮的列数为k的方案数。
dp转移方程很好想,但是dp代表式比较难,一般看见这种二维的就先固定一维再管另一维。
#include<iostream>
#include<cstdio>
using namespace std;
const int MOD=9999973;
long long dp[110][110][110];
int main(void){
int n,m,i,j,k;
scanf("%d%d",&n,&m);
dp[0][0][0]=1;
for(i=1;i<=n;i++){
for(j=0;j<=m;j++){
for(k=0;k<=m-j;k++){
dp[i][j][k]=dp[i-1][j][k];
if(k-1>=0)
dp[i][j][k]=(dp[i][j][k]+((j+1)*dp[i-1][j+1][k-1])%MOD)%MOD;
if(j-1>=0)
dp[i][j][k]=(dp[i][j][k]+((m-(j-1)-k)*dp[i-1][j-1][k])%MOD)%MOD;
if(k-2>=0)
dp[i][j][k]=(dp[i][j][k]+((j+2)*(j+1)/2*dp[i-1][j+2][k-2])%MOD)%MOD;
if(j-2>=0)
dp[i][j][k]=(dp[i][j][k]+((m-j-k+2)*(m-j-k+1)/2*dp[i-1][j-2][k])%MOD)%MOD;
if(k-1>=0&&j>=1)
dp[i][j][k]=(dp[i][j][k]+(j*(m-j-k+1)*dp[i-1][j][k-1])%MOD)%MOD;
}
}
}
long long ans=0;
for(j=0;j<=m;j++){
for(k=0;k<=m-j;k++){
//cout<<j<<" "<<k<<" "<<dp[n][j][k]<<"\n";
ans=(ans+dp[n][j][k])%MOD;
}
}
cout<<ans<<"\n";
return 0;
}