2021蓝桥杯I题国际象棋

状压dp
dp[i][j][k][p]第i行状态为j第i-1行状态为k下了p个棋子的方案数
dp[i][j][k][p]可由所有的合法的即q的状态不与j和k冲突，j和k的状态不冲突的dp[i-1][q][j][p-st[q]]转移而来
初始值为dp[0][0][0][0]=1;这样第一行所有合法状态能从dp[0][0][0][0]=1转移而来

#include <bits/stdc++.h>
#define int long long
#include <set>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define fep(i,a,b) for(int i=b;i>=a;i--)
#define prf(x) printf("%lld\n",x)
#define deprf(x) printf("[%lld]\n",x)
#define ddprf(x,y) printf("[%lld,%lld]\n",x,y)
#define scf(x) scanf("%lld",&x)
using namespace std;
const int maxn=3e3+10;
const int mod=1e9+7;
int dp[200][100][100][30];
int st[maxn];
signed main()
{
	int n,m,kk;
	scf(n);scf(m);scf(kk);
	int N=(1<<n)-1;
	rep(i,0,N)
	{
		st[i]=st[i>>1]+(i&1);
	}
	dp[0][0][0][0]=1;
	rep(i,1,m)rep(j,0,N)rep(k,0,N)
	{
		if(((j>>2)&k)||((j<<2)&k))continue;
		rep(p,0,N)
		{
			if(((p>>2)&k)||((p<<2)&k))continue;
			if(((p>>1)&j)||((p<<1)&j))continue;
			fep(q,st[j]+st[k]+st[p],kk)
			{
				dp[i][p][k][q]=(dp[i][p][k][q]+dp[i-1][k][j][q-st[p]])%mod;
			}
		}	
	}
	int ans=0;
	rep(i,0,N)rep(j,0,N)
	{
		if(((j>>2)&i)||((j<<2)&i))continue;
		ans=(ans+dp[m][i][j][kk])%mod; 
		
		 
	}
	prf(ans);
   return 0;
}