洛谷 P1896 [SCOI2005]互不侵犯King

时间:2022-10-19 05:33:45

题目描述

在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。

输入输出格式

输入格式:

只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)

输出格式:

所得的方案数

输入输出样例

输入样例#1:
3 2
输出样例#1:
16
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int n,k,i,j,w,m;//n*n的棋盘k个国王
int t[];
int f[][][];//f[i][j][k]表示i行j状态放置k个国王的方案数
int check(int x,int y)//x在上,y在下 ,1为满足条件,0为不满足条件
{
if ((x&y)!=) return ;//如果x和y中有上下都是1情况,返回0;
if ((x&(x<<))>) return ;//如果x转成二进制后有相邻的1返回0;
if ((y&(y<<))>) return ; //如果y转成二进制后有相邻的1返回0;
if ((y&(x<<))>) return ;//如果x与y的左下方有冲突,返回0;
if ((x&(y<<))>) return ;//如果x与y的右下方有冲突,返回0;
return ;//否则就返回1
}
int main()
{
cin>>n>>k;
m=(<<n)-;//2^k-1
for (i=;i<=m;i++)//求t数组
t[i]=t[i>>]+(i&);//将i转成2进制后1的个数,为何要用位运算:省时间!
f[][][]=;//初始值
for (i=;i<=n;i++)//枚举行
for (j=;j<=m;j++)//枚举本行状态
for (w=;w<=m;w++)//枚举上一行状态
if (check(w,j)==)//如果这种放置方法合法
{
for (int kkk=t[j]+t[w];kkk<=k;kkk++)//t[j]+t[w]为这两行的国王数,k为一共放置的国王数
f[i][j][kkk]=f[i][j][kkk]+f[i-][w][kkk-t[j]/*减去本行国王数之后剩下的*/];//加进去
}
int sum=;
for (i=;i<=m;i++)
sum+=f[n][i][k];//统计
printf("%d",sum);//输出
return ;
}