AcWing 1114棋盘问题 DFS(+排列组合)
up 主视频链接:DFS正确入门方式 | DFS + 递归与递推习题课(下) | 一节课教你爆搜!哔哩哔哩bilibili
题目如下:
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。
要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放 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>
来记录所有可以摆放棋子的点,然后只用对它们进行有条件的排列组合选择呢?