我天生不喜欢Dp就算你是状压DP…
Description
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
Input
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
Output
方案数。
Sample Input
3 2
Sample Output
16
HINT
Source
其实这是我的第一道状压DP…
设f[i][j][k]为到了第i行已经放了k个国王,当前行状态为k
二进制表示国王的放置位置.
然后可以进行
O(nk22n)
转移
最后求一个sum就是答案
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 1024
#define LL long long
using namespace std;
int n,k;
LL ans,f[15][100][MAXN];
inline bool check(int x,int y)
{
return !((x&(y<<1))||(x&(y>>1))||(x&y));
}
inline bool Check(int x)
{
return !((x&(x<<1))||(x&(x>>1)));
}
inline int get(int x)
{
int ret=0;
for (int i=x;i;i>>=1)
if (i&1) ret++;
return ret;
}
int main()
{
scanf("%d%d",&n,&k);
int Max=(1<<n)-1;
for (int i=0;i<=Max;i++)
if (Check(i)) f[1][get(i)][i]=1;
for (int i=2;i<=n;i++)
for (int j=0;j<=k;j++)
for (int l=0;l<=Max;l++)
if (Check(l)&&j>=get(l))
{
int tmp=get(l);
for (int t=0;t<=Max;t++)
if (Check(t)&&check(l,t)) f[i][j][l]+=f[i-1][j-tmp][t];
}
for (int i=0;i<=Max;i++)
if (Check(i)) ans+=f[n][k][i];
printf("%lld\n",ans);
}