La problemática del agua

El flujo a través (BFS)

de entrada de la muestra:

5 5
1 2 2 3 5
3 2 3 4 4
2 4 5 3 1
6 7 1 4 5
5 1 1 2 4

salida:

7

Esta pregunta no es difícil, búsqueda BFS dos veces bien, una vez que el Océano Pacífico, el Océano Atlántico una vez, y luego atraviesan el conjunto de etiquetas sería capaz de acumular. Rub durante mucho tiempo no volver a hacerlo error muy básico, se olvidó de inicializar la matriz de las etiquetas. . . O con suficiente cuidado.

#include<bits/stdc++.h>
using namespace std;
int maze[155][155],a[155][155],b[155][155],vis[155][155];
int n,m,ans;
int to[4][2]={1,0,-1,0,0,1,0,-1};
inline bool check(int x,int y){
	if(x>=0&&x<n&&y>=0&&y<m&&vis[x][y]==0)return true;
	return false;
}
struct node{
	int x,y,h;
}now;
inline void bfs(){
	queue<node>q;
	memset(vis,0,sizeof(vis));
	if(ans==1){//ans为1时把太平洋的点全部推入队列
		for(int i=0;i<n;i++){
			q.push(node{i,0,maze[i][0]});
			vis[i][0]=1;
		}
		for(int i=0;i<m;i++){
			q.push(node{0,i,maze[0][i]});
			vis[0][i]=1;
		}
	}else{//不然为大西洋
		for(int i=0;i<n;i++){
			q.push(node{i,m-1,maze[i][m-1]});
			vis[i][m-1]=1;
		}
		for(int i=0;i<m;i++){
			q.push(node{n-1,i,maze[n-1][i]});
			vis[n-1][i]=1;
		}
	}
	while(!q.empty()){//基本bfs操作
		now=q.front();
		q.pop();
		if(ans==1){
			a[now.x][now.y]=1;
		}else{
			b[now.x][now.y]=1;
		}
		for(int i=0;i<4;i++){
			int xx=now.x+to[i][0];
			int yy=now.y+to[i][1];
			if(check(xx,yy)&&now.h<=maze[xx][yy]){
				vis[xx][yy]=1;
				q.push(node{xx,yy,maze[xx][yy]});
			}
		}
	}
}
int main(){
	while(scanf("%d %d",&n,&m)!=EOF){
		for(int i=0;i<n;i++){
			for(int j=0;j<m;j++){
				scanf("%d",&maze[i][j]);
			}
		}
		memset(a,0,sizeof(a));//标记一定要清空，一直错在没初始化
		memset(b,0,sizeof(b));
		ans=1;
		bfs();
		ans=2;
		bfs();
		int sum=0;
		for(int i=0;i<n;i++){
			for(int j=0;j<m;j++){
				if(a[i][j]>0&&b[i][j]>0){
					sum++;
				}
			}
		}
		printf("%d\n",sum);
	}
	return 0;
}