bzoj 2936: [Poi1999]降 水 （floodfill算法）

2936: [Poi1999]降 水

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 55   Solved: 37
[ Submit][ Status][ Discuss]

Description

有这样一块土地，它可以被划分成N×M个正方形小块，每块面积是一平方英寸，第i行第j列的小块可以表示成P(i,j)。这块土地高低不平，每一小块地P(i,j)都有自己的高度H(i,j)（单位是英寸）。

一场倾盆大雨后，这块地由于地势高低不同，许多低洼地方都积存了不少降水。假如你已经知道这块土地的详细信息，你能求出它最多能积存多少立方英寸的降水么？

 

Input

输入文件第一行有两个数，N，M（1<=N, M <=100），表示土地的规模是N×M平方英寸。

以下有N行，每行有M个整数，表示每块地的高低（每个整数在[1,10000]内，以英寸为单位）。

 

Output

输出文件只有一行，一个数，表示土地中最多能积存多少立方英寸的水。

 

Sample Input

3 6
3 3 4 4 4 2
3 1 3 2 1 4
7 3 1 6 4 1

Sample Output

5

HINT

Source

[ Submit][ Status][ Discuss]

题解：floodfill算法。

对于矩形最外围的点必然是存不住水的，那我们可以从边界开始扩展。考虑什么样的地方能够积水，一定是他周围的格子都比他高，而最大的存水量，一定是根据他周围最低的格子确定的。

所以我们每次从队列中选取积水量最小的点，用它积水的高度更新它周围的高度，因为每次都是从小的开始更新，所以格子第一次被更新，一定是被他周围最低的格子更新，那么这一定是这个格子的积水高度。所以我们可以保证每个点只遍历一次，时间复杂度O(n*m*log(nm)).

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<queue>
#define pa pair<int,int>
#define N 103
using namespace std;
int n,m;
int vis[N][N],a[N][N],h[N][N];
int x[10]={0,1,0,-1},y[10]={1,0,-1,0};
struct data{
	int x,y,num;
	bool operator < (const data &a) const {
	  return num>a.num;
	}
};
priority_queue<data> p;
int main()
{
	freopen("a.in","r",stdin);
	scanf("%d%d",&n,&m);
	memset(vis,0,sizeof(vis));
	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) {
	  		vis[i][j]=1;
	  		p.push((data){i,j,h[i][j]});
		  }
	  }
	while (!p.empty()){
		data a=p.top(); p.pop();
		int now=a.num; 
		for (int i=0;i<4;i++)
		 {
		 	int nowx=a.x+x[i]; int nowy=a.y+y[i];
		 	if (nowx<0||nowx>n||nowy<0||nowy>m||vis[nowx][nowy]) continue;
			h[nowx][nowy]=max(h[nowx][nowy],now);
			vis[nowx][nowy]=1;
			p.push((data){nowx,nowy,h[nowx][nowy]}); 
		 }
	}
    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);
}