GYM - 101572E Emptying the Baltic

这题网上其实有很多种理解，但是最终的做法其实本质都是用了优先队列和bfs来完成。

参考：

https://blog.csdn.net/yz467796454/article/details/78250172

这里用优先队列的原因是：

如果在抽水机的深度为4的情况下，有一个方格（深度是5）和抽水机并非直接相连，而是有一个格子阻隔，比如阻隔的格子有两个，一个是深度为5，一个是3，那么本来是可以从5这个格子流入4的水的，但是如果是深度为3的点先bfs到了，就只能流过来3的水，而且因为他被vis标记过了，所以不会再扫描这里了，所以这样得到的不是最优解。所以要用优先队列，优先搜索深度最大的点来出队列。

 

注意点：

（1）

由于深度大于0的点根本就不需要搜索，所以一开始就标记好，可以节省很多时间。

 

（2）

在第一种情况下，如果由深度大的池子2流入深度小的池子1，只能留过高度为1的水量，而且这一个坑（2号）的深度也要变成更浅的深度，因为如果是

由3->2->1，充其量最终能流到1的水的深度为3，而不是4，所以2号池子在遍历玩压入队列的时候深度也要改成3（也可以把地图的深度改成3，因为这个格子最多在bfs时用上两次，而修改后只占其中的后一次）

在第二种情况下，由2--》1, 2的水可以全部流过去

 

坑点：

优先队列的自定义重载运算符的顺序本来是>的，但是在写的时候应该写成小于号！！！！

 

代码：

//E
#include<stdio.h>
#include<string.h>
#include<queue>
#include<vector>
#include<math.h>
#include<stdlib.h>
#include<functional>
using namespace std;
#define MAXN 510
int Map[MAXN][MAXN];
bool vis[MAXN][MAXN];
int dx[8]={-1,-1,0,1,1,1,0,-1};
int dy[8]={0,1,1,1,0,-1,-1,-1};
long long ans;
int w,h;

struct Node
{
    int x,y;
    int w;
    friend bool operator <(Node a,Node b)
    {
        return a.w<b.w;
    }
}pos;

bool Judge(int x,int y)
{
    if(x>=1&&x<=h&&y>=1&&y<=w)
        if(!vis[x][y])
            return true;
    return false;
}

void bfs()
{
    int i;
    priority_queue<Node>pq;
    Node now,next,temp;
    int x,y;
    int dep;

    while(!pq.empty())
        pq.pop();
    pq.push(pos);
    vis[pos.x][pos.y]=1;
    ans=abs(Map[pos.x][pos.y]);
    while(!pq.empty())
    {
        now=pq.top();
        pq.pop();
        now.w=-now.w;
        for(i=0;i<8;i++)
        {
            next=now;
            next.x+=dx[i];
            next.y+=dy[i];
            x=next.x;
            y=next.y;
            next.w=Map[x][y];
            if(Judge(x,y))
            {
                vis[x][y]=true;
                if(abs(next.w)>abs(now.w))
                    dep=abs(now.w);
                else
                    dep=abs(next.w);
                next.w=dep;
                ans+=dep;
                pq.push(next);
                temp=pq.top();
            }
        }
    }
}

int main()
{
    int i,j;

    while(scanf("%d%d",&h,&w)!=EOF)
    {
        memset(Map,0,sizeof(Map));
        memset(vis,0,sizeof(vis));
        for(i=1;i<=h;i++)
        {
            for(j=1;j<=w;j++)
            {
                scanf("%d",&Map[i][j]);
                if(Map[i][j]>=0)
                    vis[i][j]=true;
            }
        }
        scanf("%d%d",&pos.x,&pos.y);
        pos.w=Map[pos.x][pos.y];
        bfs();
        printf("%I64d\n",ans);
    }
    return 0;
}