2018.10.05 bzoj1801: [Ahoi2009]chess 中国象棋(状压dp)

版权声明:随意转载哦......但还是请注明出处吧: https://blog.csdn.net/dreaming__ldx/article/details/82946286

传送门
dp好题。
定义状态 f ( i , j , k ) f(i,j,k) 表示前i行,有j行放一个棋子,k行放两个棋子。
然后分当前放 0 / 1 / 2 0/1/2 个棋子转移。
代码:

#include<bits/stdc++.h>
#define mod 9999973
#define ll long long
using namespace std;
int n,m;
ll f[105][105][105];
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+j<=m;++k){
				f[i][j][k]=f[i-1][j][k];
				if(k){
					(f[i][j][k]+=f[i-1][j+1][k-1]*(j+1)%mod)%=mod;
					if(k>=2)(f[i][j][k]+=f[i-1][j+2][k-2]*((j+2)*(j+1)/2)%mod)%=mod;
					(f[i][j][k]+=f[i-1][j][k-1]*(m-j-k+1)%mod*j%mod)%=mod;
				}
				if(j>=2)(f[i][j][k]+=f[i-1][j-2][k]*((m-j-k+2)*(m-j-k+1)/2)%mod)%=mod;
				if(j)(f[i][j][k]+=f[i-1][j-1][k]*(m-j-k+1)%mod)%=mod;
			}
		}
	}
	ll ans=0;
	for(int i=0;i<=m;++i)for(int j=0;j+i<=m;++j)(ans+=f[n][i][j])%=mod;
	printf("%lld",ans);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/dreaming__ldx/article/details/82946286