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