AcWing 1114棋盘问题 DFS（+排列组合）

AcWing 1114棋盘问题 DFS（+排列组合）

up 主视频链接：DFS正确入门方式 | DFS + 递归与递推习题课(下) | 一节课教你爆搜!哔哩哔哩bilibili

题目链接：1114. 棋盘问题 - AcWing题库

题目如下：

---

在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。

要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放 k 个棋子的所有可行的摆放方案数目 C。

输入格式

输入含有多组测试数据。

每组数据的第一行是两个正整数 n,k，用一个空格隔开，表示了将在一个 n*n 的矩阵内描述棋盘，以及摆放棋子的数目。当为-1 -1时表示输入结束。

随后的 n 行描述了棋盘的形状：每行有 n 个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。

输出格式

对于每一组数据，给出一行输出，输出摆放的方案数目 C（数据保证 C<2^{31}）。

数据范围

n\leq 8,k\leq n

输入样例：

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

输出样例：

2
1

---

初看此题就想到了经典的八皇后问题——那么就用 dfs 来做。

我是这样定义 dfs 函数的：

void dfs(int count,int res,int x,int y)

• 首先我用 count 记录已经摆放了几个棋子，并且基于此搭建了 dfs 的框架。

• 然后第二步，我加了一个 res 来记录还剩多少个棋盘区域可以用于摆放棋子。

ps : 这样做可以使用 if(count+res<k) return ; 来剪枝。

写好了以上两步，提交，WA。调试，发现很严重的问题——重复枚举了！（例如把 {(1,2),(2,1)} 和 {(2,1),(1,2)} 判定成两种方案了）

为了避免重复，可以引入 x 和 y 来记录前一个点的横纵坐标，然后之后就只用枚举 (x,y) 之后的格子啦！

如何达到这样的效果？

直接 for(int i=x;i<=n;++i) 嵌套 for(int j=y;j<=n;++j) ？

不对！

在 i=x 这一排要从 j=y 这一列开始枚举

而在之后的排，j 要从第一列开始枚举！

于是有了以下的 dfs （更多细节见注释）：

void dfs(int count,int res,int x,int y) // 已选count个 剩余res个 
{
    if(count+res<k) return ;
    if(count>=k) // 已选k个,cnt++并且函数return ;
    {
        cnt++;
        return ;
    }
    
    //这里使用了st_r和st_c两个数组来记录行(row)和列(column)是否可用
    for(int i=x;i<=n;++i)
    {
        if(st_r[i]) continue; // 单独把判断行、列是否可用的判断放外面，节省时间！
        for(int j=1;j<=n;++j)
        {
            if(i==x && j<y) continue; // 这里的处理前文已讲
            if(st_c[j]) continue; 
            if(maze[i][j]=='#')
            {
                st_r[i]=1;st_c[j]=1;
                res--;                  // 这里为什么只对res单独-1而且不恢复现场？
                dfs(count+1,res,i,j);   // 因为过了这个村，那么此次旅行就再也没有这个村了;而count是还能回溯的！
                st_r[i]=0;st_c[j]=0; // 恢复现场
            }
        }
    }
}

完整代码奉上：

#include<iostream>
#include<cstring>
​
using namespace std;
const int N=9;
​
char maze[N][N];
​
int n,k,cnt,num;
​
bool st_r[N],st_c[N];
void dfs(int count,int res,int x,int y) // 已选count个 剩余res个 
{
    if(count+res<k) return ;
    if(count>=k)
    {
        cnt++;
        return ;
    }
    
    for(int i=x;i<=n;++i)
    {
        if(st_r[i]) continue;
        for(int j=1;j<=n;++j)
        {
            if(i==x && j<y) continue;
            if(st_c[j]) continue; 
            if(maze[i][j]=='#')
            {
                st_r[i]=1;st_c[j]=1;res--;
                dfs(count+1,res,i,j);
                st_r[i]=0;st_c[j]=0;
            }
        }
    }
}
int main()
{
    while(1)
    {
        cin>>n>>k;
        if(n==-1 && k==-1) break;
        memset(maze,-1,sizeof(maze));
        cnt=0;num=0;
        
        for(int i=1;i<=n;++i)
        {
            for(int j=1;j<=n;++j)
            {
                cin>>maze[i][j];
                if(maze[i][j]=='#')
                    ++num;
            }
        }
        dfs(0,num,1,1);
        cout<<cnt<<endl;
    }
    
    return 0;
}

---

题后思考：整个题目又像是一道加了一些限制的 排列组合 问题，是否可以用 pair<int,int> 来记录所有可以摆放棋子的点，然后只用对它们进行有条件的排列组合选择呢？