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