棋盘问题（dfs+回溯）

题目链接：

[kuangbin带你飞]专题一 简单搜索 - Virtual Judgehttps://vjudge.net/contest/65959#problem/A

思路：

回溯模板题，类似leetcode的N皇后问题，不过没有斜行限制。另外，可以用vis数组来记录遍历过的列，减少遍历次数。

代码：

 

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <stack>
using namespace std;
const int maxn = 9;
char mp[maxn][maxn];
bool vis[maxn]; // 储存该列是否已经放过棋子
int n, k; // 棋盘变长，棋子数量
int ans; // 可填入方法数

void dfs(int x, int cnt){ // x代表行数，cnt代表当前填入棋子数
    if(cnt == k) ans++;
    for(int i=x; i<n; i++){ // 选择行
        for(int j=0; j<n; j++){ // 选择列
            if(mp[i][j] == '#' && vis[j] == false){ // 满足题意，可以在这里放棋子
                vis[j] = true;
                dfs(i+1, cnt+1);
                vis[j] = false; // 回溯
            }
        }
    }
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    while(cin >> n >> k){
        if(n == -1) break;
        ans = 0;
        memset(vis, false, sizeof(vis));
        for(int i=0; i<n; i++){
            scanf("%s", mp[i]);
        }
        dfs(0, 0);
        cout << ans << endl;
    }
}