洛谷 P2051 [AHOI2009]中国象棋

题目描述

这次小可可想解决的难题和中国象棋有关，在一个N行M列的棋盘上，让你放若干个炮（可以是0个），使得没有一个炮可以攻击到另一个炮，请问有多少种放置方法。大家肯定很清楚，在中国象棋中炮的行走方式是：一个炮攻击到另一个炮，当且仅当它们在同一行或同一列中，且它们之间恰好 有一个棋子。你也来和小可可一起锻炼一下思维吧！

输入输出格式

输入格式：

一行包含两个整数N，M，之间由一个空格隔开。

输出格式：

总共的方案数，由于该值可能很大，只需给出方案数模9999973的结果。

输入输出样例

输入样例#1：

1 3

输出样例#1：

7

说明

样例说明

除了3个格子里都塞满了炮以外，其它方案都是可行的，所以一共有2*2*2-1=7种方案。

数据范围

100%的数据中N和M均不超过100

50%的数据中N和M至少有一个数不超过8

30%的数据中N和M均不超过6

动态规划，需要知道任意一列或一行最多放两个炮，炮数相同的列或行是没有区别的，剩下的看代码吧。

#include<iostream>
#include<cstdio>
using namespace std;
const int N=105;
const long long mod=9999973;
int n,m;
long long ans,f[N][N][N];//前i行，有j列放了一个炮，有k列放了两个炮。 
long long calc(int x)
{
	return x*(x-1)/2;
}
int main()
{
	scanf("%d%d",&n,&m);
	f[0][0][0]=1;
	for(int i=1;i<=n;i++)
		for(int j=0;j<=m;j++)
			for(int k=0;k<=m-j;k++)
			{
				f[i][j][k]=f[i-1][j][k];//不放 
				if(j>0)
					f[i][j][k]+=f[i-1][j-1][k]*(m-j+1-k),f[i][j][k]%=mod;//在空列放一个 
				if(k>0)
					f[i][j][k]+=f[i-1][j+1][k-1]*(j+1),f[i][j][k]%=mod;//在有一个棋子的一列放一个 
				if(j>1)
					f[i][j][k]+=f[i-1][j-2][k]*calc(m-j-k+2),f[i][j][k]%=mod;//在两个空列各放一个 
				if(k>1)
					f[i][j][k]+=f[i-1][j+2][k-2]*calc(j+2),f[i][j][k]%=mod;//在有一个棋子的两列各放一个 
				if(k>0)
					f[i][j][k]+=f[i-1][j][k-1]*j*(m-j-k+1),f[i][j][k]%=mod;//在有一个棋子的一列和一个空列各放一个 
			}
	for(int i=0;i<=m;i++)
		for(int j=0;j<=m;j++)
			ans+=f[n][i][j],ans%=mod;
	printf("%lld\n",ans);
	return 0;
}