这一题的话,我们搜索的时候,边界条件就是棋子的数目等于k,这时候方案数加一然后return。
如果超过行号的最大值,我们也返回,不过不做任何操作,这也是一个边界条件。
如果 r 行的 j 个位置是棋盘并且这一列没有放入棋子,我们就放入一颗棋子,该列标记为1,已放棋子数加一,然后访问下一行。
这个递归结束以后,我们就把它的标记清空,放入棋子数减一,还原之前的情况,然后递归搜索这一行不放棋子的方案数。
#include <iostream>
#include <cstring>
using namespace std;
int map[10][10];
int vis[10];
int n, k, ans, cnt;
void DFS(int r)
{
if (cnt==k) {
ans++;
return;
}
if (r>=n)
return;
for (int j = 0; j < n;j++) {
if (!vis[j]&&map[r][j]) {
cnt++;//在这一行放下棋子
vis[j] = 1;
DFS(r + 1);
vis[j] = 0;
cnt--;//复原这一行
}
}
DFS(r + 1);//在上一行不放棋子的情况下搜索下一行
}
int main()
{
char ch;
ios::sync_with_stdio(false);
while (cin >> n >> k&&n!=-1&&k!=-1) {
memset(map, 0, sizeof(map));
memset(vis, 0, sizeof(vis));
for (int i = 0; i < n;i++) {
for (int j = 0; j < n;j++) {
cin >> ch;
if (ch=='#')
map[i][j] = 1;
}
}
ans = cnt = 0;
DFS(0);
cout << ans << endl;
}
return 0;
}