【ybtoj】山峰和山谷

山峰和山谷

---

题目描述

给定一个n*n的网格状地图，每个方格（i,j）有一个高度$w_i{,}_j$。如果两个方格有公共顶点，则它们是相邻的。
定义山峰山谷如下：
均由地图上的一个联通块组成。
所有方格高度都相同。
周围的方格（即不属于山峰或山谷但与山峰或山谷相邻的格子）高度均大于山谷的高度，或小于山峰的高度。
求地图内山峰和山谷的数量。特别的，如果整个地图方格的高度均相同，则整个地图即是一个山谷，也是一个山峰。

输入格式

第一行一个正整数n，表示地图的大小。
接下来n行，每行n个整数表示地图。第i行有n个正整数$w_i{,}_1,w_i{,}_2,w_i{,}_3,...,w_i{,}_n$，表示地图第i行格子的高度。

输出格式

输出一行两个整数，分别表示山峰和山谷的数量。

输入样例

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 1

解题思路

考虑BFS，每次从一个暂未被访问过的点出发进行BFS

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
int n,a[1010][1010],a2[8]= {-1,-1,0,1,1,1,0,-1},a3[8]= {0,1,1,1,0,-1,-1,-1},t,s1=0,s2=0;
bool b[1010][1010],t1,t2;
struct f {
	int x,y;
}k,kt,kd;
queue<f>c;
void bfs() {
	t1=0,t2=0;
	while(!c.empty()) 
	{
		kt=c.front();
		c.pop();
		for(int i=0;i<=7;i++) 
		{
			kd.x=kt.x+a2[i];
			kd.y=kt.y+a3[i];
			if(!(kd.x>=1&&kd.x<=n&&kd.y>=1&&kd.y<=n))
			 continue;
			if(!b[kd.x][kd.y]&&a[kd.x][kd.y]==t) 
			{
				b[kd.x][kd.y]=1;
				c.push(kd);
			} 
			else 
			{
				if(a[kd.x][kd.y]<t)t1=1;
				if(a[kd.x][kd.y]>t)t2=1;
			}
		}
	}
	return;
}
int main() 
{
	cin>>n;
	for(int i=1; i<=n; i++)
	 for(int j=1; j<=n; j++)
	  cin>>a[i][j];
	for(int i=1; i<=n; i++)
	 for(int j=1; j<=n; j++) 
	 {
	    if(!b[i][j]) 
		{
			b[i][j]=1;
			t=a[i][j];
			k.x=i,k.y=j;
			c.push(k);
			bfs();
			if(!t1&&!t2)
			 s1++,s2++;
			if(t1&&!t2)//判断山顶或是山峰
			 s1++;
			if(!t1&&t2)
			 s2++;
		}
	}
	cout<<s1<<" "<<s2;
	return 0;
}

---

谢谢阅读