Red and Black（HDU1312）---图的搜索

题目链接

题目描述

有一间长方形的房子，地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上，只能向相邻的黑色瓷砖移动。请写一个程序，计算你总共能够到达多少块黑色的瓷砖。

输入格式

包括多个数据集合。每个数据集合的第一行是两个整数W和H，分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中，每行包括W个字符。每个字符表示一块瓷砖的颜色，规则如下:

• ‘.’：黑色的瓷砖；
• ‘#’：白色的瓷砖；
• ‘@’：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次；

当在一行中读入的是两个零时，表示输入结束。

输出格式

对每个数据集合，分别输出一行，显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。

输入样例

6 9
…#.
…#
…
…
…
…
…
#@…#
.#…#.
0 0

输出样例

45

思路

典型的搜索题，注意为避免搜索越界需要在其边界加上“篱笆”。使用BFS或者DFS搜索均可，时间复杂度O(nm)。

源程序

DFS版本

#include <bits/stdc++.h>
#define MAXN 25
using namespace std; 
int n,m,sx,sy,cnt;
int mov[4][2]={{-1,0},{0,1},{1,0},{0,-1}};
char g[MAXN][MAXN];
bool used[MAXN][MAXN];
void dfs(int x,int y)
{
	for(int i=0;i<4;i++){
		int xx=x+mov[i][0];
		int yy=y+mov[i][1];
		if(g[xx][yy]=='.'&&!used[xx][yy]){
			used[xx][yy]=true;
			cnt++;
			dfs(xx,yy);
		}
	}
	
}
int main()
{
	while(1){
		cin>>m>>n;
		if(n==0&&m==0)break;
		memset(used,false,sizeof(used));
		for(int i=1;i<=n;i++)	//输入数据 
			for(int j=1;j<=m;j++){
				cin>>g[i][j];
				if(g[i][j]=='@')sx=i,sy=j;
			}
		for(int i=0;i<=m+1;i++)g[0][i]=g[n+1][i]='#';	//建立上下篱笆
		for(int i=0;i<=n+1;i++)g[i][0]=g[i][m+1]='#';	//建立左右篱笆
		used[sx][sy]=true,cnt=1;	//初始化 
		dfs(sx,sy);
		cout<<cnt<<endl;
	}
	return 0;
}

BFS版本

#include <bits/stdc++.h>
#define MAXN 25
using namespace std; 
struct Node{
	int x,y;
};
int n,m,sx,sy,cnt;
int mov[4][2]={{-1,0},{0,1},{1,0},{0,-1}};
char g[MAXN][MAXN];
bool used[MAXN][MAXN];
void bfs(int x,int y)
{
	queue<Node> q;
	memset(used,false,sizeof(used));//初始化 
	used[sx][sy]=true,cnt=1;	
	q.push(Node{sx,sy});
	while(!q.empty()){
		Node tmp=q.front();
		q.pop();
		for(int i=0;i<4;i++){
			int xx=tmp.x+mov[i][0];
			int yy=tmp.y+mov[i][1];
			if(g[xx][yy]=='.'&&!used[xx][yy]){
				used[xx][yy]=true;
				cnt++;
				q.push(Node{xx,yy});
			}
		}
	}
} 
int main()
{
	while(1){
		cin>>m>>n;
		if(n==0&&m==0)break;
		for(int i=1;i<=n;i++)	//输入数据 
			for(int j=1;j<=m;j++){
				cin>>g[i][j];
				if(g[i][j]=='@')sx=i,sy=j;
			}
		for(int i=0;i<=m+1;i++)g[0][i]=g[n+1][i]='#';	//建立上下篱笆
		for(int i=0;i<=n+1;i++)g[i][0]=g[i][m+1]='#';	//建立左右篱笆
		bfs(sx,sy);
		cout<<cnt<<endl;
	}
	return 0;
}