poj 1321 棋盘问题
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
INPUT
输入含有多组测试数据,每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
OUTPUT
对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)。
Sample Input
2 1
#.
.#
4 4
…#
…#.
.#…
#…
-1 -1
Sample output
2
1
类似八皇后的题目,DFS入门级别
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int N = 10;
char a[N][N];
bool row[10];//标记每列是否有棋子
int n,k,cot=0,sum=0;
void dfs(int x) //对第x行搜索
{
if(k==cot){
sum++;
return;
}//结束条件,已经放入要求个数
if(x>n)return;//超出边界
for(int j=1; j<=n; j++)
if(a[x][j]=='#'&&row[j]==0){ //满足条件即可摆棋子
cot++;
row[j]=1;
dfs(x+1);//搜索下一行
row[j]=0;//搜索之后回溯剪枝,去除标记
cot--;
}
dfs(x+1);//直接搜索下一行
}
int main()
{
while(cin>>n>>k&&n!=-1&&k!=-1){
sum=cot=0;
memset(row,0,10);//初始化
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++){
cin>>a[i][j];
}
dfs(1);//从第一行开始搜索
printf("%d\n",sum);
}
return 0;
}
算法标签:DFS 回溯
版权声明:本文为qq_53485165原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。