【NCPC2017-2018-gym101572-E.Emptying the Baltic 】优先队列

Emptying the Baltic

题目链接：

https://codeforces.com/gym/101572/problem/A

Description

Input

Output

Sample Input

3 3
-5 2 -5
-1 -2 -1
5 4 -5
2 2

Sample Output

10

题意

给你一个n*m的水池，每个格子都有一个相对于水平面的高度，如果为负数说明在水平面之下。

之后在某个点挖了一个井，现在问有多少水会流到井里，当A的高度<=B的高度时，B的水就会流到A那里。

题解

首先最初只有起点周围的四个地点会向起点流水。

设起点周围的某个点下标为i，高度为$h_i$,起点的高度为$x$。

如果h[i]>x,那么说明i点最低可以取到$h_i$的水。

如果h[i]<x,那么说明i点最低可以取到x的水。

而且之后水池中的每个点一定是被他周围四个点中能取得最深的点去更新深度。

这个过程就好像dij的过程，每次找出当前最优的点，去更新他能到的点，每个点只会被更新一次。

过程中要注意当一个点的高度<0时才说明里面有水可以流。

复杂度$nmlog(nm)$

代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<queue>
#include<vector>
using namespace std;
#define Fi first
#define Se second
#define dbg(x) cout<<#x<<" = "<<x<<endl
#define dbg2(x1,x2) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<endl
#define dbg3(x1,x2,x3) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<" "<<#x3<<" = "<<x3<<endl
#define ok cout<<"OK"<<endl
typedef long long ll;
typedef pair<int,int> pii;
const int INF = 0x3f3f3f3f;
const int maxn =1e3+10;
int a[maxn][maxn],n,m;
int vis[maxn][maxn];
int dp[maxn][maxn];
int d[8][2]={1,0,-1,0,0,1,0,-1,1,-1,-1,-1,1,1,-1,1};
struct data
{
    int x,y;
    int val;
    data(){}
    data(int xx,int yy,int vv)
    {
        x=xx;
        y=yy;
        val=vv;
    }
    const bool operator < (const data &y)const
    {
        return val>y.val;
    }
};
int main()
{
    //freopen("E-4.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    int xx,yy;
    scanf("%d%d",&xx,&yy);
    if(a[xx][yy]>=0) return 0*puts("0");
    priority_queue<data> pq;
    pq.push(data(xx,yy,a[xx][yy]));
    vis[xx][yy]=1;
    dp[xx][yy]=a[xx][yy];
    while(!pq.empty())
    {
        data tp=pq.top();
        pq.pop();
        int x=tp.x;
        int y=tp.y;
        int val=tp.val;
        for(int i=0;i<8;i++)
        {
            int tx=x+d[i][0];
            int ty=y+d[i][1];
            if(tx<1||tx>n||ty<1||ty>m||a[tx][ty]>=0||vis[tx][ty]) continue;
            vis[tx][ty]=true;
            dp[tx][ty]=max(a[tx][ty],val);
            pq.push(data(tx,ty,dp[tx][ty]));
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) if(vis[i][j]) ans=ans+max(0,0-dp[i][j]);
    printf("%lld\n",ans);
    return 0;
}