c++ DFS练习 红与黑

红与黑是DFS的一题经典练习题，比较简单，尽可能自己打出代码

题目原题如下：
There is a rectangular room, covered with square tiles. Each tile is colored either red or black. A man is standing on a black tile. From a tile, he can move to one of four adjacent tiles. But he can’t move on red tiles, he can move only on black tiles.
Write a program to count the number of black tiles which he can reach by repeating the moves described above.
Input
The input consists of multiple data sets. A data set starts with a line containing two positive integers W and H; W and H are the numbers of tiles in the x- and y- directions, respectively. W and H are not more than 20.
There are H more lines in the data set, each of which includes W characters. Each character represents the color of a tile as follows.
‘.’ - a black tile
‘#’ - a red tile
‘@’ - a man on a black tile(appears exactly once in a data set)
Output
For each data set, your program should output a line which contains the number of tiles he can reach from the initial tile (including itself).

Sample Input

6 9
…#.
…#
…
…
…
…
…
#@…#
.#…#.
11 9
.#…
.#.#######.
.#.#…#.
.#.#.###.#.
.#.#…@#.#.
.#.#####.#.
.#…#.
.#########.
…
11 6
…#…#…#…
…#…#…#…
…#…#…###
…#…#…#@.
…#…#…#…
…#…#…#…
7 7
…#.#…
…#.#…
###.###
…@…
###.###
…#.#…
…#.#…
0 0

Sample Output

45
59
6
13

翻译（谷歌）如下：

有一个长方形的房间，覆盖着正方形的瓷砖。每个磁贴都被着色为红色或黑色。一个人站在黑色的瓷砖上。他可以从一个图块移动到四个相邻图块之一。但是他不能在红色瓷砖上移动，只能在黑色瓷砖上移动。

编写一个程序，通过重复上述动作来计算他可以到达的黑色瓷砖的数量。
输入项
输入包含多个数据集。数据集从包含两个正整数W和H的行开始； W和H分别是x和y方向上的图块数。 W和H不超过20。

数据集中还有H行，每行包含W个字符。每个字符表示瓦片的颜色，如下所示。

‘.’ -黑色瓷砖
‘＃’-红色方块
‘@’-黑色瓷砖上的一个人（在数据集中只出现一次）
输出量
对于每个数据集，您的程序应输出一行，其中包含他可以从初始磁贴（包括其自身）达到的磁贴数量。
样本输入（如上）
样本输出（如上）

应为本题数据很水，所以直接暴搜就能AC

读入
应为有多组数据，所以用while语句读入
dfs在运行过程中需要记录访问的节点，防止多次访问同个点，陷入死循环，因此用一个vis[x][y]数组表示（x，y）这个点是否能 或 已被访问
***********************************注意，需要包括不能访问的节点
代码如下

while(cin>>n>>m)
{
	if(n==0&&m==0)
	break;
	ans=1;  //本行下文讲
	for(i=1;i<=m;i++)
	for(j=1;j<=n;j++)
	{
		cin>>c[i][j];
		if(c[i][j]=='@')
		{
			x=i;y=j;//x，y记录起始点
			vis[i][j]=1;//防止起始点多次被访问
		}
		else
		if(c[i][j]=='#')
		vis[i][j]=1;//表示无法访问
}

DFS
首先定义s1,s2数组，用来模拟上·下·左·右·

nt  s1[5]={0,1, -1, 0, 0},
	s2[5]={0,0, 0, 1, -1};
	//       上 下 左  右 

再进行深度搜索

inline void dfs(int x,int y)
{
	int i,j,xx,yy;
	for(i=1;i<=4;i++)
	{
		xx=x+s1[i];
		yy=y+s2[i];
		if(xx>0&&xx<=m&&yy>0&&yy<=n&&vis[xx][yy]==0)
		{
			ans++;//ans记录可以被访问点的个数，即答案
			vis[xx][yy]=1;//将可以被访问的点变成不能访问，表示已被访问，防止死循环
			dfs(xx,yy);//进行下一层搜索
		}
	}
}

注意！！！ 应为本题的初始点也要被访问，而vis数组中将初始点标为不能访问是因为ans定义是负值为1，表示起始点已被访问一次，所以ans=1而不是0，而本题的DFS也是基于这一点写的。纯属个人习惯 ~ ~~ ：）

最后记得清空数组；） 本程序用了最笨的清空方法233

最后放上完整代码
完整程序：

//#include
#include
using namespace std;
int n,m,x,y,z,ans=1;
char c[1010][1010];
int vis[1010][1010]={0};
int s1[5]={0,1,-1,0,0},s2[5]={0,0,0,1,-1};
inline void dfs(int x,int y)
{
	int i,j,xx,yy;
	for(i=1;i<=4;i++)
	{
		xx=x+s1[i];
		yy=y+s2[i];
		if(xx>0&&xx<=m&&yy>0&&yy<=n&&vis[xx][yy]==0)
		{
			ans++;
			vis[xx][yy]=1;
			dfs(xx,yy);
		}
	}
}
int main()
{
	int i,j;
	while(cin>>n>>m)
	{
		if(n==0&&m==0)
		break;
		ans=1;
		for(i=1;i<=m;i++)
		for(j=1;j<=n;j++)
		{
			cin>>c[i][j];
			if(c[i][j]=='@')
			{
				x=i;y=j;
				vis[i][j]=1;
			}
			else
			if(c[i][j]=='#')
			vis[i][j]=1;
		}
		dfs(x,y);
		cout<<ans<<endl;
		
		for(i=1;i<=m;i++)
		for(j=1;j<=n;j++)
		vis[i][j]=0;
	}
	return 0;
}

总结
简单的DFS模板题，需要灵活掌握。

本文原创，仅供参考
2019.12.21