DFS习题-洛谷P1683入门

洛谷P1683 入门

#include<iostream>
using namespace std;
int w,h,res;
char a[25][25];
int flag[25][25];
int fx[]={-1,0,0,1};
int fy[]={0,-1,1,0};

void dfs(int x,int y){
	for(int i=0;i<4;i++){
		int nx=x+fx[i];
		int ny=y+fy[i];
		if(nx<1||nx>h||ny<1||ny>w) continue;
		if(flag[nx][ny]==1) continue;
		if(a[nx][ny]=='#') continue;
	//	if(a[nx][ny]!='.') continue;
		flag[nx][ny]=1;
		res++;
		dfs(nx,ny);
		// 注：不要回溯
	}

}
int main(){
	cin>>w>>h;
	for(int i=1;i<=h;i++){
		for(int j=1;j<=w;j++){
			cin>>a[i][j];
		}
	}
	for(int i=1;i<=h;i++){
		for(int j=1;j<=w;j++){
			if(a[i][j]=='@'){
				flag[i][j]=1;
				dfs(i,j);
			}
		}
	}
	cout<<res+1<<endl;
	return 0;
}