【题目描述】
在N×N
(
1
≤
N
≤
9
)
(1≤N≤9)
(1≤N≤9)的棋盘里面放K
(
0
≤
K
≤
N
2
)
(0≤K≤N^2)
(0≤K≤N2)个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
【输入格式】
一行,两个数N,K
【输出格式】
一个数,总方案数
S a m p l e I n p u t Sample~Input Sample Input
3 2
S a m p l e    O u t p u t Sample\;Output SampleOutput
16
【题意分析】
状压DP基础题。(这种奇奇怪怪的放置问题而且数据奇小无比基本都是状压DP)
先处理出一行之内可能的放法,放在sit[]
里面,并计算出这种放法一行之内有几个
K
i
n
g
King
King,放在King[]
里面。
d p [ i ] [ j ] [ k ] dp[i][j][k] dp[i][j][k]表示第i行,当前行状态为j,而且放了k个 K i n g King King的时候的方法总数。
//实际上状态是 s i t [ j ] sit[j] sit[j]。已经预处理。
一个nb操作:__builtin_popcount ()
可以求出一个数二进制形式里面有几个1。
第一行dp数组初始化处理: 如果这种放法的
K
i
n
g
King
King数量没有超过要求,那么
d
p
[
1
]
[
i
]
[
K
i
n
g
[
i
]
]
=
1
;
dp[1][i][King[i]] = 1;
dp[1][i][King[i]]=1;
与上同理,状态其实是 s i t [ i ] sit[i] sit[i]。
然后正式开始DP,分别枚举行号i,该行状态j,上一行状态k。
先判断是否有冲突:
sit[j] & sit[k](中间)
(sit[j] << 1) & sit[k](左边)
sit[j] & (sit[k] << 1)(右边)
然后再枚举
K
i
n
g
King
King数量,如果没有超过总数,就加上去。
最后答案是把每行的各种状态的方法都加起来
(为什么每行的都要加?举个例子,你有可能第1行就把国王用完了,也有可能是下面的任何行。)
Code:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cctype>
#include<cstring>
#include<vector>
#include<algorithm>
#define int long long
using namespace std;
int dp[10][15000][80],sit[800000],King[80000];
signed main (){
int n,king;
scanf ("%lld%lld",&n,&king);
int SIT = (1 << n) - 1,tot = 0;
for (register int i = 0;i <= SIT;i++){
if ((i << 1) & i)continue;
sit[++tot] = i;
King[tot] = __builtin_popcount (i);
}
for (register int i = 1;i <= tot;i++)
if (King[i] <= king)dp[1][i][King[i]] = 1;
//以上为初始化
for (register int i = 2;i <= n;i++)
for (register int j = 1;j <= tot;j++)
for (register int k = 1;k <= tot;k++){
if (sit[j] & sit[k])continue;
if ((sit[j] << 1) & sit[k])continue;
if (sit[j] & (sit[k] << 1))continue;
//不冲突
for (register int l = 1;l <= king;l++){
if (King[j] + l > king)continue;
dp[i][j][King[j]+l] += dp[i-1][k][l];//更新
}
}
int ans = 0;
for (register int i = 1;i <= n;i++)
for (register int j = 1;j <= tot;j++)
ans += dp[i][j][king];
printf ("%lld",ans);
return 0;
}