棋盘问题 dfs

棋盘问题
在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放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

题目链接：http://poj.org/problem?id=1321
这个题其实和八皇后问题比较类似
八皇后链接：https://blog.csdn.net/weixin_42475137/article/details/86795536
可以采取和八皇后类似的策略。但是也有不一样的地方，八皇后的棋子个数和棋盘边长的相等的，而这个题是不一样的，所以需要一个额外的变量来存放棋子总数。

#include<iostream>
#include<cstring>
using namespace std;
int n, k;
char map[9][9];   //储存棋盘
int book[9] = { 0 }; //记录每行棋盘的第几个位置有棋子，没有棋子的话标记为0
int number = 0;   //记录种数
bool judge(int x,int n) { //判断当前位置放棋子是否合理，
 for (int i = x - 1; i > 0; i--) { //因为是按行搜索，所以不用判断同一行
  if (book[i] == n) {
   return false;
  }
 }
 return true;
}
void dfs(int cur,int num) {  //给第cur行安排棋子，num代表当前已经排放的棋子总数
 if (num==k) {    //如果摆好k个棋子，种数加一
  number++;
  return;     //返回
 }
 if (cur > n) {    //如果当前行数大于棋盘边，返回
  return;
 }
 for (int i = 0; i <= n; i++) { //给当前行的第i个位置摆放棋子
  if (i == 0) {    //如果不摆放，安排下一行，总数不变
   dfs(cur + 1, num);
  }
  if (map[cur][i] == '#'&&judge(cur, i)) { //如果棋盘可以摆放，且棋子不冲突
   book[cur] = i;     //摆好棋子
   dfs(cur + 1, num + 1);   //安排下一行，总数加一
   book[cur] = 0;     //函数递归完，要把之前的棋子拿走
  }
 }
}
int main() {
 cin >> n >> k;
 while (n!=-1||k!=-1) {
  for (int i = 1; i <= n; i++) { //输入棋盘
   for (int j = 1; j <= n; j++) {
    cin >> map[i][j];
   }
  }
  dfs(1, 0); //从第一行开始摆放，此时棋子总数为0
  cout << number << endl;  //输出种数
  cin >> n >> k;
  memset(book, 0, sizeof(int) * 9); //重置数组和number
  number = 0;
 }
 return 0;
}