bzoj 1102: [POI2007]山峰和山谷Grz （floodfill）

1102: [POI2007]山峰和山谷Grz

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 747   Solved: 385
[ Submit][ Status][ Discuss]

Description

　　FGD小朋友特别喜欢爬山，在爬山的时候他就在研究山峰和山谷。为了能够让他对他的旅程有一个安排，他想
知道山峰和山谷的数量。给定一个地图，为FGD想要旅行的区域，地图被分为n*n的网格，每个格子(i,j) 的高度w(
i,j)是给定的。若两个格子有公共顶点，那么他们就是相邻的格子。（所以与(i,j)相邻的格子有(i?1, j?1),(i?1
,j),(i?1,j+1),(i,j?1),(i,j+1),(i+1,j?1),(i+1,j),(i+1,j+1)）。我们定义一个格子的集合S为山峰（山谷）当
且仅当：1.S的所有格子都有相同的高度。2.S的所有格子都联通3.对于s属于S，与s相邻的s’不属于S。都有ws > 
ws’（山峰），或者ws < ws’（山谷）。你的任务是，对于给定的地图，求出山峰和山谷的数量，如果所有格子
都有相同的高度，那么整个地图即是山峰，又是山谷。

Input

　　第一行包含一个正整数n，表示地图的大小（1<=n<=1000）。接下来一个n*n的矩阵，表示地图上每个格子的高
度。(0<=w<=1000000000)

Output

　　应包含两个数，分别表示山峰和山谷的数量。

Sample Input

输入样例1
5
8 8 8 7 7
7 7 8 8 7
7 7 7 7 7
7 8 8 7 8
7 8 8 8 8
输入样例2
5
5 7 8 3 1
5 5 7 6 6
6 6 6 2 8
5 7 2 5 8
7 1 0 1 7

Sample Output

输出样例1
2 1
输出样例2
3 3

HINT

Source

[ Submit][ Status][ Discuss]

题解：floodfill 

山峰就是一片相同的值连成且周围没有比当前高度高的点。

我们可以把每个点的高度扔到一个大根堆中，然后依次弹出队列中的点，如果当前点没有访问过，答案+1，然后从这个点向四周扩展，每次只扩展高度小于等于当前高度的点，并打访问标记。

求山谷的话，将高度取反，再重复上述过程即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define N 1003
using namespace std;
int n,m,vis[N][N],a[N][N],h[N][N],ans;
int x[10]={-1,-1,-1,0,0,1,1,1},y[10]={-1,0,1,-1,1,-1,0,1};
struct data{
	int x,y,v;
	bool operator < (const data &a) const{
	  return v<a.v;
	}
};
priority_queue<data> p,q;
void bfs(int x1,int y1,int v)
{
	 q.push((data){x1,y1,v});
	 while(!q.empty()) {
	 	data a1=q.top(); q.pop();
	 	for (int i=0;i<8;i++)
	 	 {
	 	 	int nowx=a1.x+x[i];
	 	 	int nowy=a1.y+y[i];
	 	 	if (nowx<=0||nowx>n||nowy<=0||nowy>n||vis[nowx][nowy]) continue;
	 	 	if (a[a1.x][a1.y]>=a[nowx][nowy]) {
			    vis[nowx][nowy]=1;
	 	 	//h[nowx][nowy]=max(h[nowx][nowy],h[a1.x][a1.y]);
	 	 	//if (h[nowx][nowy]==h[a1.x][a1.y]) {
	 	 		q.push((data){nowx,nowy,h[nowx][nowy]});
			  }
		  }
	 }
}
int main()
{
	freopen("a.in","r",stdin);
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	 for (int j=1;j<=n;j++)
	  scanf("%d",&a[i][j]),p.push((data){i,j,a[i][j]}),h[i][j]=a[i][j];
	memset(vis,0,sizeof(vis));
	while (!p.empty()){
		data a=p.top(); p.pop();
		if (!vis[a.x][a.y]) {
			vis[a.x][a.y]=1;
			bfs(a.x,a.y,a.v); 
			ans++;
		}
	}
	printf("%d ",ans);
	int ans=0;
	memset(vis,0,sizeof(vis));
	for (int i=1;i<=n;i++)
	 for (int j=1;j<=n;j++)
	  a[i][j]=-a[i][j],p.push((data){i,j,a[i][j]}),h[i][j]=a[i][j];
	while (!p.empty()){
		data a=p.top(); p.pop();
		if (!vis[a.x][a.y]) {
			vis[a.x][a.y]=1;
			bfs(a.x,a.y,a.v); 
			ans++;
		}
	}
	printf("%d\n",ans);
}