详见:http://blog.csdn.net/clover_hxy/article/details/52956172
久闻Floodfill算法
对于矩形最外围的点必然是存不住水的,那我们可以从边界开始扩展。考虑什么样的地方能够积水,一定是他周围的格子都比他高,而最大的存水量,一定是根据他周围最低的格子确定的。
所以我们每次从队列中选取积水量最小的点,用它积水的高度更新它周围的高度,因为每次都是从小的开始更新,所以格子第一次被更新,一定是被他周围最低的格子更新,那么这一定是这个格子的积水高度。所以我们可以保证每个点只遍历一次,时间复杂度O(n*m*log(nm)).
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<queue>
using namespace std;
const int N=105;
int n,m;
int vst[N][N],a[N][N],h[N][N];
const int dx[]={0,1,0,-1};
const int dy[]={1,0,-1,0};
struct abcd{
int x,y,h;
abcd(int x,int y,int h):x(x),y(y),h(h) { }
bool operator < (const abcd &B) const {
return h>B.h;
}
};
priority_queue<abcd> Q;
int main(){
freopen("t.in","r",stdin);
freopen("t.out","w",stdout);
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++){
scanf("%d",&a[i][j]); h[i][j]=a[i][j];
if (i==1||j==1||i==n||j==m)
vst[i][j]=1,Q.push(abcd(i,j,h[i][j]));
}
while (!Q.empty()){
abcd u=Q.top(); Q.pop();
for (int k=0;k<4;k++){
int x=u.x+dx[k],y=u.y+dy[k];
if (x<0 || x>n || y<0 || y>m || vst[x][y]) continue;
h[x][y]=max(h[x][y],u.h);
vst[x][y]=1;
Q.push(abcd(x,y,h[x][y]));
}
}
int Ans=0;
for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) Ans+=h[i][j]-a[i][j];
printf("%d\n",Ans);
}