POJ 1321 棋盘问题 --- DFS

时间:2023-01-30 10:06:52

  POJ 1321

  题目大意:给定一棋盘,在其棋盘区域放置棋子,需保证每行每列都只有一颗棋子。 (注意 .不可放 #可放)

  解题思路:利用DFS,从第一行开始依次往下遍历,列是否已经放置棋子用一个数组标记,直至放完要求的棋子数。

/*POJ 1321 棋盘问题 --- DFS*/
#include <cstdio>
#include <cstring> int n, k, cnt;
bool visit[]; //标记列的访问状态
char mapp[][]; /*从第r行开始正确放置p个棋子*/
void dfs(int r, int p){
if (p == ){
++cnt;
return;
}
if (r >= n || n-r < p) //剪枝 --> n-r >= p才能成功放置
return;
for (int j = ; j < n; ++j){
if (mapp[r][j] == '#' && !visit[j]){
visit[j] = ; //在r行j列放一枚棋子
dfs(r + , p - );
visit[j] = ;
}
}
dfs(r + , p); //由于棋子数小于行数, 可以第r行不放棋子,从下一行开始
} int main()
{
while (scanf("%d%d", &n, &k) == ){
if (n == - && k == -)
break;
cnt = ;
memset(visit, , sizeof visit);
for (int i = ; i < n; ++i){
scanf("%s", mapp[i]);
}
dfs(, k);
printf("%d\n", cnt);
} return ;
}