[SCOI2005]互不侵犯题解
——HM
题目描述
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
输入输出格式
输入格式:
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
输出格式
所得的方案数
输入输出样例
输入样例#1
3 2
输出样例#1
16
题解
看一眼大概有两种思路:搜索和DP,搜索的话好像没有什么方法,似乎都会TLE,也有可能我太蒻了,基本可以肯定用动态规划来实现。再看一眼数据范围,只有九,基本可以确定使用状压DP来完成。
教程中有讲状压DP的套路,这里不再赘述,只要知道状压DP说白了就是用二进制来进行枚举。
可以将每一行情况用1和0表示,1指此处有国王,0指此处没有国王,如10101表示第1、3、5个格子有国王,第2、4个格子中没有,因为非有即无,所以每行都能类似的用如此的01串来表示,即为二进制表示数。上面所说10101及可表示为十进制的数字21。
用数组dp[i][j][k]来表示第i行的状态为j时共摆了k个国王,此时的总方案数,显然答案就是将所有k=K的dp值加起来。
如何判断是否能吃到彼此呢,每一行中可以用位运算来表示,即
if ( !(now&( now<<1 )) )
若该语句返回真,则说明其左右不相邻。其实这个句子较好理解,将当前状态(now)向左移一位后与原状态做与运算,很显然就是将当前状态和此状态前一位比较是否都为1,若都为1,则说明有两个国王相邻了,故不成立。上下,斜着的上下左右也是一个意思,只是将两个不同的状态中进行位运算。
易得f[i][j][k]+=f[i−1][now[i-1]][k-king[k]]
编程是再注意一些细节即可。一定要开long long啊!炸了好几次70.
附略丑的标程:
#include <bits/stdc++.h>
using namespace std;
const int SIZE=1<<15;
const int INF=0x3f3f3f3f;
#define ll long long
bool check(ll x,ll y);
ll n,m,tot=0,ans=0;
ll now[SIZE],dp[15][SIZE][105],ks[SIZE];
int main()
{
scanf("%lld%lld",&n,&m);
for (int i=0;i<=(1<<n)-1;i++)
if ( !( (i<<1)&i ) ){
tot++;
now[tot]=i;
ll x=i;
while (x){
ks[tot]+=x%2;
x/=2;
}
}
for (int i=1;i<=tot;i++)
if (ks[i]<=m)
dp[1][i][ks[i]]=1;
for (int i=2;i<=n;i++)
for (int j=1;j<=tot;j++)
for (int k=1;k<=tot;k++){
if (check(j,k)) continue;
for (int p=1;p<=m;p++){
if (ks[j]+p>m) continue;
dp[i][j][ks[j]+p]+=dp[i-1][k][p];
}
}
for (int i=1;i<=n;i++)
for (int j=1;j<=tot;j++)
ans+=dp[i][j][m];
printf("%lld",ans);
return 0;
}
bool check(ll x,ll y)
{
if (now[x]&now[y]) return true;
else if (now[x]&(now[y]<<1)) return true;
else if ((now[x]<<1)&now[y]) return true;
else return false;
}