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> 来记录所有可以摆放棋子的点，然后只用对它们进行有条件的排列组合选择呢？