GYM - 101572E Emptying the Baltic 优先队列+BFS

题目链接: GYM - 101572 E

题目大意

n*m的地图, 每个格子有一个海拔高度, 海拔<0的地方有水, 现在在(x, y)最深处放一个抽水机, 问最多能抽多少水, 水只能从高出往低处流

思路

一开始我想到用普通队列BFS, 但是因为水可以往八个方向流, 如果先出队列的不出最深的一个点的话, 原本能流过去的水可能就流不过去了, 所以用优先队列, 保证先出队列的是当前已处理的最深的节点

代码

Accepted 124 ms 5200 KB

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

const int maxn = 600;
int s[maxn][maxn], n, m;
bool vist[maxn][maxn];
int dx[] = {1, -1, 0, 0, -1, -1, 1, 1};
int dy[] = {0, 0, -1, 1, -1, 1, -1, 1};
struct P
{
    int x, y;
    P(int i, int j):x(i), y(j) {}
    P(){}
    bool operator < (const P&b) const
    {
        return s[x][y] > s[b.x][b.y];
    }
};

int main()
{
    cin >> n >> m;
    for(int i=0; i<n; ++i)
        for(int j=0; j<m; ++j) scanf("%d", &s[i][j]);
    int x, y;
    scanf("%d%d", &x, &y);
    --x, --y;

    priority_queue<P> que;
    que.push(P(x, y));
    vist[x][y] = 1;
    while(!que.empty())
    {
        P now = que.top(); que.pop();
        for(int i=0; i<8; ++i)
        {
            int nx = now.x + dx[i];
            int ny = now.y + dy[i];
            if(0<=nx && nx<n && 0<=ny && ny<m && !vist[nx][ny])
            {
                vist[nx][ny] = 1;
                s[nx][ny] = max(s[now.x][now.y], s[nx][ny]);
                que.push(P(nx, ny));
            }
        }
    }
    long long ans = 0;
    for(int i=0; i<n; ++i) for(int j=0; j<m; ++j) if(s[i][j] < 0) ans += -s[i][j];
    cout << ans << endl;
    return 0;
}