题目描述
给定一个 n×n的网格状地图,每个方格 (i,j)有一个高度 wij 。如果两个方格有公共顶点,则它们是相邻的。
定义山峰和山谷如下:
- 均由地图上的一个连通块组成;
- 所有方格高度都相同;
- 周围的方格(即不属于山峰或山谷但与山峰或山谷相邻的格子)高度均大于山谷的高度,或小于山峰的高度。
求地图内山峰和山谷的数量。特别地,如果整个地图方格的高度均相同,则整个地图既是一个山谷,也是一个山峰。
输入
第一行一个整数 n(2≤n≤1000),表示地图的大小。
接下来 n行每行 n个整数表示地图。第 i行有 n个整数 wi1,wi2,…,win(0≤wij≤1 000 000 000),表示地图第 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
样例输入 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
样例输出 2
3 3
C++
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef pair<int,int> PII; // 定义 地图坐标的数据类型
const int N = 1010 , M = N * N;
int g[N][N]; // 地图
PII q[M]; // 自定义queue
int dx[8]={-1,-1,-1,0,0,1,1,1}; // x偏移量
int dy[8]={-1,0,1,-1,1,-1,0,1}; // y偏移量
bool st[N][N];
int n;
bool higher,lower; // 山峰还是山谷
void bfs(int x,int y){
// tt queue 尾 hh queue 头
int tt=0,hh=0;
q[0]={x,y}; //将初始点放入queue
st[x][y]=true;
while(tt>=hh)
{
PII t=q[hh++];
for(int i=0;i<8;i++)
{
int a=t.first+dx[i],b=t.second+dy[i];
if(a>=0&&a<n&&b>=0&&b<n)
{
if(g[a][b]!=g[t.first][t.second])
{
// 如果联通块的周围都不比当前那个点高 该点就是 谷底
// 如果连通块的周围都不比当前那个点低 该点就说山峰
// higher 和 lower 初始都为 false
// 如果有一点比联通块中的点改 那 higher就变为true 那该点就不是peak
// 如果有一点比连通块中的点低 那 lower就为 true 那该连通图就不是valley
if(g[a][b]>g[t.first][t.second]) higher=true;
else lower=true;
}
else if(!st[a][b]) // 地图上相同 并且没被遍历 就扩展
{
st[a][b]=true;
q[++tt]={a,b};
}
}
}
}
}
int main(){
scanf("%d",&n);
// 输入地图
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
scanf("%d",&g[i][j]);
// 定义山谷 和 山峰数量的变量
int valley = 0,peak = 0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
if(!st[i][j]) // 如果地图没被遍历过
{
higher=false;
lower=false;
bfs(i,j);
if(!higher) {
peak++;
}
if(!lower) {
valley++;
}
}
printf("%d %d\n",peak,valley);
return 0;
}