AcWing1114 棋盘问题（DFS）

题目链接

知识点：DFS

搜索思路：限制条件只有列与行只放一个，定义col数组记录该列是否被占用，dfs的参数x表示正在填第几行，所以不用考虑一行有多个，枚举的对象为x行的每一列，因为每一行可填可不填所以枚举完每一列后，再在for循环外后面加一个dfs(x+1);代表不取x行，直接搜索下一行。

代码：

#include <iostream>
#include <vector>
using namespace std;
typedef long long ll;
int n, k, sum;
int col[50] = {0};
char a[10][10];
vector<int> resX;
void dfs(int x)
{
    if (k == 0 && resX.size() > 0) 
    //判断棋子是否取完，要与判断是否超过棋盘行数分开，因为可能还未到达棋盘最后一行棋子就取完了
    {
        sum++;
        return;
    }
    if (x > n) //防止不断的不取棋子死循环，设置边界
        return;
    for (int i = 1; i <= n; i++)
    {
        int y = i;
        if (a[x][y] == '.' || col[y] != 0)
            continue;
        col[y] = 1, k--, resX.push_back(x);
        dfs(x + 1);
        col[y] = 0, k++, resX.erase(resX.end() - 1);
    }
    dfs(x + 1); //关键：表示不取这一行棋子直接进入下一行，我就是因为把它写在上面for循环里卡了半天，位置很重要
}
int main()
{
    while (~scanf("%d%d", &n, &k))
    {
        sum = 0;
        if (n == -1 && k == -1)
            break;
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n; j++)
                cin >> a[i][j];
        dfs(1);
        cout << sum << endl;
    }
    return 0;
}