计蒜客-dfs(深搜)-红与黑

计蒜客-dfs(深搜)-红与黑

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

输入格式第一行是两个整数 W 和 H，分别表示 x 方向和 y 方向瓷砖的数量。
W和 H 都不超过 202020。在接下来的 H行中，每行包括 W 个字符。每个字符表示一块瓷砖的颜色，

规则如下：
1）’.’：黑色的瓷砖；
2）’#’：白色的瓷砖；
3）’@’：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
输出格式输出一行，显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。

样例输入：
6 9
. . . . # .
. . . . . #
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
#@…#
.#…#.

样例输出：
45

#include <iostream>
#include<cstdio>
using namespace std;
char maze[40][40];
int vis[40][40];
int w,h,i,j,x,y;
int ans=1;
int dir[4][2]={{-1,0},{0,-1},{1,0},{0,1}};
bool in (int x,int y){
 return x>=0&&x<h&&y>=0&&y<w;
}
void dfs(int x, int y){
    vis[x][y]=1;
    for(int i=0;i<4;i++){
     int tx=x+dir[i][0];
     int ty=y+dir[i][1];
     if(!vis[tx][ty]&&maze[tx][ty]=='.'&&in(tx,ty)){
         ans++;
         dfs(tx,ty);
  }
 }
}
int main(){
    cin>>w>>h;
        for(i = 0; i < h; i++){
            cin>>maze[i];
            for(j = 0; j < w; j++){
                if(maze[i][j] == '@'){ 
                    x=i;y=j;
                }
            }
        }
        dfs(x,y);
    cout<<ans<<endl;
    return 0;
}

bfs写法

#include<bits/stdc++.h>
using namespace std;
const int N=25;
int vis[N][N];
char maze[N][N];
typedef pair<int,int> PII;
#define x first
#define y second
int n,m,fx,fy;
int dx[]={-1,0,1,0};
int dy[]={0,1,0,-1};

bool check(int x,int y){
	return x<0||x>=n||y<0||y>=m;
} 
int bfs(int fx,int fy){
	vis[fx][fy]=1;
	queue<PII> q;
	q.push({fx,fy});
	int ans=0;
	while(q.size()){
		PII t=q.front();
		q.pop(); 
		ans++;
		for(int i=0;i<4;i++){
			int a=t.x+dx[i];
			int b=t.y+dy[i];
			if(check(a,b)||vis[a][b]==1||maze[a][b]=='#') continue;
			vis[a][b]=1;
			q.push({a,b});
		}	
	} 
	return ans;
}
int main(){
	while(cin>>m>>n,n||m){
		memset(vis,0,sizeof vis);
		for(int i=0;i<n;i++){
			for(int j=0;j<m;j++){
				cin>>maze[i][j];
				if(maze[i][j]=='@'){
					fx=i;
					fy=j;
				}
			}
		}
	cout<<bfs(fx,fy)<<endl;
	}
	return 0;
}