poj 2227 (优先队列、floodfill算法)

题意:一个n*m矩形区域,其中每个1*1都是有高度的矩形块,高低起伏,求最多储水量。(边界不能储水)

分析:水桶效应。先将边界装入优先队列中(高度越小越优先),并标记为已访问。看队首元素四周未访问过的点,1、如果该点不比队首低,则将它加入队列,标记为已访问,即它变成了新的边界。2、该点比队首低,意味着该点可以储水,更新res值,同时将它加入队列中,但是它的高度为原队首元素的高度,即以它为边界的点不能超过这个高度,同时将该点标记为已访问。

注意:优先队列取队首元素是q.top(),不是一般队列的q.front()。

代码:

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
#define ll long long
const int N = 300+10;
int mp[N][N],vis[N][N],n,m;
int to[4][2]={-1,0,1,0,0,-1,0,1};
struct nd{
    int x,y,h;
    friend bool operator < (nd a,nd b) {
        return a.h>b.h;
    }
};
priority_queue<nd> q;

void init(){
    while(!q.empty()) q.pop();
    nd tp;
    memset(vis,0,sizeof(vis));
    for(int i=0;i<n;i++) {
        for(int j=0;j<m;j++) {
            if( i==0 || j==0 || i==n-1 || j==m-1 ) {
                vis[i][j]=1;
                tp.x=i,tp.y=j,tp.h=mp[i][j];
                q.push(tp);
            }
        }
    }
}

void bfs() {
    init();
    ll ans=0;
    nd t,tp;
    while(!q.empty()) {
        tp=q.top();
        q.pop();
        for(int i=0;i<4;i++) {
            int xx=tp.x+to[i][0],yy=tp.y+to[i][1];
            if( 0<=xx&&xx<n && 0<=yy&&yy<m && !vis[xx][yy]) {
                vis[xx][yy]=1;
                t.x=xx,t.y=yy,t.h=mp[xx][yy];
                if(t.h<tp.h) ans+=(tp.h-t.h),t.h=tp.h;
                q.push(t);
            }
        }
    }
    cout<<ans<<endl;
}

int main() {
    while(cin>>m>>n) {
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                cin>>mp[i][j];
        bfs();
    }
}

猜你喜欢

转载自blog.csdn.net/tianwei0822/article/details/80330143