Gym 101572E Emptying Baltic【优先队列+BFS】

题目大意：

给出一个N*M的矩阵，然后在点（x，y）处设立一个抽水机，每个点有一个数代表海拔高度，负数就有水，水往低处流，问能够抽多少水。

思路：

直接Bfs一下即可，模拟拓展出去的过程。

Ac代码：

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
int a[505][505];
int vis[505][505];
int fx[8]={0,0,1,-1,1,-1,1,-1};
int fy[8]={1,-1,0,0,1,-1,-1,1};
struct node
{
    int x,y,val;
    bool friend operator <(node a,node b)
    {
        return a.val>b.val;
    }
}now,nex;
int n,m;
int sx,sy;
void Bfs()
{
    memset(vis,0,sizeof(vis));
    priority_queue<node>s;
    now.x=sx;now.y=sy;now.val=a[sx][sy];
    vis[sx][sy]=-a[sx][sy];s.push(now);
    while(!s.empty())
    {
        now=s.top();
        s.pop();
        for(int i=0;i<8;i++)
        {
            nex.x=now.x+fx[i];
            nex.y=now.y+fy[i];
            nex.val=a[nex.x][nex.y];
            if(nex.x>=1&&nex.x<=n&&nex.y>=1&&nex.y<=m&&vis[nex.x][nex.y]==0)
            {
                vis[nex.x][nex.y]=1;
                a[nex.x][nex.y]=max(a[nex.x][nex.y],a[now.x][now.y]);
                s.push(nex);
            }
        }
    }
    __int64 output=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(a[i][j]>=0)continue;
            else output+=-a[i][j];
        }
    }
    printf("%I64d\n",output);
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
            {
                scanf("%d",&a[i][j]);
            }
        }
        scanf("%d%d",&sx,&sy);
        Bfs();
    }
}
/*
5 5
-8 -8 -8 -8 -8
-8 -1 -1 -1 -8
-8 -1 -2 -1 -8
-8 -1 -1 -1 -8
-8 -8 -8 -8 -8
*/