题意:给定一个N * N的棋盘,放置K个国王,使他们互相攻击不到对方,共有多少种方案。国王可以攻击上下左右,左上左下,右上右下附近的一格,共8格。
数据范围:1<=N<=9, 0<=K<=N * N.
因为N的范围很小,而且每一行对应的一个方案可以用一个二进制数表示,所以容易想到用状压DP。又因题目有限制要用多少个国王,所以我们需要记录一下国王的个数。
于是,我们可以定义DP状态:
dp[i,j,k]表示第i行采用j方案,国王总个数为k时的总方案数。
那么如何推出状态转移方程呢?
状压DP在于枚举,我们先想想怎么枚举。
设置can[i],记录在不考虑行与行之间影响的情况下,每行可以采取的方案。那么,因为每个方案对应有用到一定个数的国王,所以再设置一个num[i],记录采用can[i]方案时用了多少个国王。
初始完can[i],num[i]后,首先初始化第一行:dp[1,can[i],num[i]]=1.
对于i ← 2 ~ N行,假如i-1行采用了can[j]方案,i行采用了can[k]方案,那么在can[k]与can[j]不冲突的情况下,不难推出:dp[i,can[k],num[k]+p]+=dp[i,can[j],p],其中0<=p<=K-num[k],即表示第i行采用can[k]方案时在前一行采用can[j]方案,且国王总个数为p的情况下推出来的。
代码如下:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=1<<9;
ll dp[10][maxn][90];
int can[maxn],num[maxn];
int count(int x)
{
int sum=0;
while(x)
{
if(x%2) sum++;
x/=2;
}
return sum;
}
int main()
{
int N,K;
cin>>N>>K;
int cnt=0,tot=1<<N;
for(int i=0;i<tot;i++)
{
if(!(i&(i<<1))) //同行各个国王不能相邻
{
int c=count(i);
if(c<=K)
{
can[++cnt]=i;
num[cnt]=c;
}
}
}
memset(dp,0,sizeof(dp));
for(int i=1;i<=cnt;i++)
dp[1][can[i]][num[i]]=1;
for(int i=2;i<=N;i++)
for(int j=1;j<=cnt;j++)
{
for(int k=1;k<=cnt;k++)
{
if(!(can[k]&can[j])&&!((can[k]<<1)&can[j])&&!((can[k]>>1)&can[j])) //相邻两行不冲突
{
for(int p=0;p<=K-num[k];p++) //枚举
dp[i][can[k]][num[k]+p]+=dp[i-1][can[j]][p];
}
}
}
ll sum=0;
for(int i=1;i<=cnt;i++) //最终答案落在sum(dp[N,can[1~cnt],K])
sum+=dp[N][can[i]][K];
cout<<sum<<endl;
}