题意:一个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();
}
}