T1T2
思路:DP+状压+预处理!
DP[i][s][k]为当前在i行,状态为s,放了k个国王的方案数!
DP[i][S][k]=∑ DP[i-1][s][k-f(s)] s是合法的上一层状态,f(p)是p的棋子数。
考虑每一次枚举,状态有2^9种=512,512*512*9*9*9 >1亿。(枚举2层状态,枚举行数,枚举更新棋子数,检查合法)。
貌似会超时吧!
于是准备预处理状态!
对于每行的状态,我们不能让一个国王的两侧有国王
先筛出符合的状态,记录在det数组中
枚举det数组中的一个状态,再枚举一个
判断两个状态作为上下行是否冲突,记录在一个布尔数组里即可
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<cmath>
#define LL long long
using namespace std;
LL n,k;
LL f[10][1<<11][90];//f[i][j][k]表示前i行 当前行状态j 棋子有k个
LL gp[1<<11];//每个状态有几个棋子
LL sum;
bool map[1<<11][1<<11];//当前状态和上一层状态是否合法
LL det[1<<11];
LL cnt=0;
void init()//预处理
{
for(LL i=0;i<=sum;i++)
{
if(!(i&(i<<1))&&!(i&(i>>1)))
{
det[++cnt]=i;
for(LL j=0;j<n;j++)
{
if(i&(1<<j))gp[cnt]++;
}
}
}
for(LL i=1;i<=cnt;i++)
for(LL j=i;j<=cnt;j++)
{
bool ok=true;
if(det[i]&det[j])ok=false;
if(det[i]&(det[j]<<1))ok=false;
if(det[i]&(det[j]>>1))ok=false;
if(ok)map[i][j]=map[j][i]=true;
}
}
int main()
{
scanf("%lld%lld",&n,&k);
sum=(1<<n)-1;
init();
f[0][1][0]=1;
for(LL i=0;i<n;i++)
{
for(LL j=1;j<=cnt;j++)
{
for(LL l=0;l<=k;l++)
{
if(f[i][j][l])
{
for(LL p=1;p<=cnt;p++)
{
if(map[j][p]&&l+gp[p]<=k)
{
f[i+1][p][l+gp[p]]+=f[i][j][l];
}
}
}
}
}
}
LL ans=0;
for(LL i=1;i<=sum;i++)ans+=f[n][i][k];
printf("%lld\n",ans);
return 0;
}