棋盘问题,dfs+回溯

点击题目
大致题意:
在一个矩阵里面,有个不规则的棋盘,#代表可以放旗子, ·代表不能放棋子
要求放k个棋子,在不同行不同列,有多少种放法

思路:
不同行不同列,可以按照行放,每一行在某一个列选出一个棋子,然后标志这一列已经被用了.
vis_col[ ]用来标记
chess[ ]表示棋盘 为1的地方表示可以放棋子

dfs(int r,int num) // r代表在第r行找一列出来,num代表这是放的第num个棋子
当 num 等于k ,结果加1
for(寻找没有被访问过的某一列)
{
继续递归寻找下一行的某一列,num+1;
}
//可能k<n,有些行不需要访问,为了考虑所有可能,不访问这一行
dfs(r+1,num)

#include<iostream>
#include<cstring>
using namespace std;


int chess[10][10];
int vis_col[10];


int n, k;
int cnt;
void dfs(int r, int num);
int main()
{
	ios::sync_with_stdio(false);
	while (cin >> n >> k){
		if (n == -1) return 0;
		char c;
		cnt = 0;
		memset(chess, 0, sizeof(chess));
		memset(vis_col, 0, sizeof(vis_col));
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= n; j++)
			{
				cin >> c;
				if (c == '#'){
					chess[i][j] = 1;
				}
			}
		dfs(1, 0);
		cout << cnt << endl;
	}
	return 0;
}
void dfs(int r, int num)  //r表示在第r行寻找某一列,num代表已经安排的棋子数
{
	if (num == k)
	{
		cnt++;
		return;
	}
	if (r > n)return;   //行数超过棋盘大小
	for (int i = 1; i <= n; i++)
	{
		if (!vis_col[i] && chess[r][i])
		{
			vis_col[i] = 1;
			dfs(r + 1, num + 1);
			vis_col[i] = 0;
		}
	}
	dfs(r + 1, num);    //难点! k<n 可能有些行不需要访问,如果只有上面的,那么会漏掉情况,只剩下,只访问了前k行,后面的没有管
		//这一行不访问,访问下一行,就可以统计所有条件,if(r>n) return,就是这种情况很多行不访问,防止溢出的
}