题目链接:
http://poj.org/problem?id=1321
思路:
这题用dfs来解 dfs函数只有一个形参,那就是行数,因为题目中说了两个棋子不能放在一个棋盘的同一行或同一列,所以,一行只能选一个能放棋子的位置来放棋子。首先定义一个数组,visited用来记录第i列是否有棋子放入,注意 是第i列。
然后定义一个全局变量way来记录已经往棋盘中放入了多少个棋子了,递归的边界条件就是way等于k;
代码如下:
#include<bits/stdc++.h>
using namespace std;
int n,k;
char a[10][10];
int visited[10];//visited数组记录在前h-1行第几列已经放入棋子了
int cont=0;//cont代表种数
int way=0;
void dfs(int h)//h代表行数 way代表选择放入棋子的个数
{
if(way==k)//当放入的棋子数等于给定需要放入的棋子数时 方案数加1
{
cont++;
return ;
}
if(h>=n)//当所搜索的行数大于棋盘时 直接返回
return ;
for(int i=0;i<n;i++)//在第h行中从第一个位置开始列举
{
if(!visited[i]&&a[h][i]=='#')//当前h-1行中第i列中已经放入了棋子 那么第h行就不用放了 因为棋子不能放在同一行 同一列
{
visited[i]=1;
way++;
dfs(h+1);
visited[i]=0;//还原 假如第0行有两个# 当从第0行的第一个#开始往下走时,假设第2行中有且只有一个#,并且不与第一行中的任意一个在同一列,那么在往下
way--;//走时就会把这个#标记 当递归返回到第一行第一个#时 i会++ 到第一行的第2个# 函数继续会继续往下走 ,到达第2行的那个#,因为在第一行第一个#那条路径中
//已经走过了第2行的那个#(因为第2行只有一个#并且这个#所在的列数不与第一行两个#重合),如果不把第一次走过第二行的#的标记消除 那么这次路径就不会再过第二行的
//这个# 那么可选路径少了 肯定不对了
}
}
dfs(h+1);
}
int main()
{
while (cin>>n>>k)
{
if(n==-1&&k==-1)
break ;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
cin>>a[i][j];
dfs(0);
cout<<cont<<endl;
memset(visited,0,sizeof(visited));
cont=0;
way=0;
}
}
代码分析:
分析一下dfs函数如何运行。首先,在主函数中先将h=0进入dfs函数(表示搜索第1行),然后用一个for循环来依次访问这一行中的每个字符,如果可以往下走,就把visited[i]标记为1(即第i列已经走过),让way加1,这样,dfs进入第2行,然后在进行上述过程,当way等于给定的k时,种数加1。假如way小于k,但是h已经大于题目给定的行数时,直接返回上一层函数,不用再访问了,因为这一行不存在。dfs的关键在于当这一层所有的字符都访问完成后,应该在再个dfs(h+1),如果没有,那么当其中一行所有的字符都不满足if(!visited[i]&&a[h][i]=='#')这个条件是,直接返回上一行,然后再从上一行的下一个字符接着进入到这一行,但是这一行没有满足条件的,所以dfs函数不可能进入到这一行的下一行,所以无法探索这一行后面行的字符,所以肯定不对。但当加了dfs(h+1)后,因为这一行所有的字符都不会满足条件,那么for循环结束后会因为dfs(h+1)这条语句进入下一行,继续访问这一行的下一行