这个是我做 kuangbin 带你飞专题训练的第一个专题的第一题
题意
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
==其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。 ==
思路
这是一道经典的深搜问题。
dfs函数只有一个形参,那就是行数,因为题目中说了两个棋子不能放在一个棋盘的同一行或同一列,所以,一行只能选一个能放棋子的位置来放棋子。首先定义一个数组a用来记录第j列是否有棋子放入.
注意 是第j列
然后定义一个全局变量num来记录已经往棋盘中放入了多少个棋子了,递归的边界条件就是num等于k.
代码如下:
#include <iostream>
using namespace std;
int a[10];
char chess[10][10];//定义棋盘
int flag=0;//flag代表摆放棋盘的方案数
int num=0;//num代表选择放入棋子的个数
int n,k;
void dfs(int r) //r代表行数
{
if(num == k) //如果放入棋子数num达到k,那么计数器(及方案数)加一,并返回。
{
flag++;
return;
}
if(r >= n) return;//当搜索的行数大于棋盘时,直接返回(判断边界)。
for(int j = 0;j < n; j++)//在r行从第一个位置开始例举
{
if(!a[j] && chess[r][j] == '#')
//当前r-1行中第j列中已经放入了棋子,那么第h行就不用放了(因为棋子不能放在同一行 同一列)如果符合条件,则该位置可放棋子。
{
a[j] = 1;
num++; //标记为已访问,棋子数加一。
dfs(r+1);//深搜下一行
a[j] = 0;//如果深搜失败或者棋子数目够了,则撤销标记。
num--;
}
}
dfs(r+1);//该行已经有棋子,深搜下一行。
}
int main(){
while(cin>>n>>k&&n != -1&&k != -1)
{
for(int i = 0; i < n; i++)
for(int j = 0 ; j < n; j++)
cin>>chess[i][j];
flag = num = 0;//别忘记每次都要初始化,这是多样例题。
dfs(0);//从第一行开始深搜。
cout<<flag<<endl;
}
return 0;
}
分析
dfs的关键在于当这一层所有的字符都访问完成后,应该在再个dfs(r+1),如果没有,那么当其中一行所有的字符都不满足if(!a[i]&&chess[h][i]==’#’)这个条件是,直接返回上一行,然后再从上一行的下一个字符接着进入到这一行,但是这一行没有满足条件的,所以dfs函数不可能进入到这一行的下一行,所以无法探索这一行后面行的字符,所以肯定不对。但当加了dfs(r+1)后,因为这一行所有的字符都不会满足条件,那么for循环结束后会因为dfs(r+1)这条语句进入下一行,继续访问这一行的下一行。