题解 | 挡住洪水

挡住洪水

https://www.nowcoder.com/practice/56e54f4c2e3c4a58abfe76dbc1da1d7e

#include<bits/stdc++.h>
using namespace std;

const int N=600;

int n,m,res=0;
char g[N][N];
bool vis[N][N];
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};

void bfs(int x,int y){
	
	queue<pair<int,int>> q;
	q.push({x,y});
	vis[x][y]=true;
	bool f_res=true;
	int area_cnt=1;
	
	while(!q.empty()){
		
		pair<int,int> pr=q.front();q.pop();
		int x=pr.first,y=pr.second;
		
		int tx,ty;
		for(int i=0;i<4;i++){
			tx=x+dx[i];
			ty=y+dy[i];
			
			if(tx<1||tx>n||ty<1||ty>m){
				f_res=false;
				continue;
			}
			if(g[tx][ty]=='*') continue;
			if(vis[tx][ty]==true) continue;
			area_cnt++;
			q.push({tx,ty});
			vis[tx][ty]=true;
		}
		
	}
	
	if(f_res==true) res+=area_cnt;
	
	return;
}

void solve(){
	
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			if(g[i][j]=='0'&&!vis[i][j]){
				bfs(i,j);
			}
		}
	}
	
	cout<<res<<endl;
	return;
	
}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			cin>>g[i][j];
		}
	}
	
	solve();

    return 0;
}


全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务