题目描述:
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 为山峰(山谷)当且仅当:
- S 的所有格子都有相同的高度。
- S 的所有格子都连通。
- 对于 s 属于 S,与 s 相邻的 s′ 不属于 S,都有 ws>ws′(山峰),或者 ws<ws′(山谷)。
- 如果周围不存在相邻区域,则同时将其视为山峰和山谷。
你的任务是,对于给定的地图,求出山峰和山谷的数量,如果所有格子都有相同的高度,那么整个地图即是山峰,又是山谷。
输入格式
第一行包含一个正整数 n,表示地图的大小。
接下来一个 n×n 的矩阵,表示地图上每个格子的高度 w。
输出格式
共一行,包含两个整数,表示山峰和山谷的数量。
数据范围
1≤n≤1000,
0≤w≤10^9
输入样例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
输出样例1:
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
样例解释
样例1:
样例2:
分析:
本题给定了一个由数字填充的正方形区域,所有连通的相同的数字构成了一个连通域,当该连通域的周围数字都小于该连通域数字时,该连通域就是一个山峰,当周围数字都大于该连通域数字时,该连通域就是一个山谷,当周围即不存在大于连通域数字又不存在小于连通域数字时,该连通域既是山峰又是山谷,题目要求我们统计山峰和山谷的数目。
我们只需要在bfs的过程中,遇见大于连通域数字就说明该连通域不是山峰,遇见小于连通域数字就说明该连通域不是山谷,用这样两个标志变量来确定该连通域的性质就可以解决问题了。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1005;
typedef pair<int,int> PII;
int g[N][N],n,res1 = 0,res2 = 0;
int dx[] = {0,0,-1,1,-1,-1,1,1};
int dy[] = {1,-1,0,0,-1,1,-1,1};
bool st[N][N];
PII q[N * N];
void bfs(int x,int y){
int hh = 0,tt = 0;
q[0] = {x,y};
st[x][y] = true;
bool flag1 = true,flag2 = true;
while(hh <= tt){
PII t = q[hh++];
for(int i = 0;i < 8;i++){
int nx = t.first + dx[i],ny = t.second + dy[i];
if(nx < 0 || nx >= n || ny < 0 || ny >= n) continue;
if(g[nx][ny] != g[t.first][t.second]){
if(g[nx][ny] < g[t.first][t.second]) flag2 = false;
else flag1 = false;
}
else if(!st[nx][ny]){
q[++tt] = {nx,ny};
st[nx][ny] = true;
}
}
}
if(flag1) res1++;
if(flag2) res2++;
}
int main(){
scanf("%d",&n);
for(int i = 0;i < n;i++){
for(int j = 0;j < n;j++){
scanf("%d",&g[i][j]);
}
}
for(int i = 0;i < n;i++){
for(int j = 0;j < n;j++){
if(!st[i][j]) bfs(i,j);
}
}
printf("%d %d\n",res1,res2);
return 0;
}