项目场景:
Flood Fill算法的应用。
问题描述
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 的矩阵,表示地图上每个格子的高度 ww。
输出格式
共一行,包含两个整数,表示山峰和山谷的数量。
数据范围
1≤n≤1000,
0≤w≤109
输入样例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:
原因分析:
在Flood fill算法统计连通块的基础上,判断一个连通块是否为山峰或山谷,需要在进行标记连通块时,判断连通块边界与相邻的位置的高度关系,并标记has_higher和has_lower的值,通过这两个参数的值来进行统计连通块中山峰和山谷的个数。
实现代码:
#include<iostream>
#include<cstring>
#include<algorithm>
#define x first
#define y second
using namespace std;
typedef pair<int,int> PII;
const int N = 1010;
PII q[N*N];
int n;
int g[N][N];
bool st[N][N];
void bfs(int sx,int sy,bool& has_higher,bool& has_lower){
int hh = 0,tt = 0;
q[0] = {sx,sy};
st[sx][sy] = true;
while(hh<=tt){
auto t = q[hh++];
//八点邻接的访问形式
for(int i=t.x-1;i<=t.x+1;i++)
for(int j = t.y-1;j<=t.y+1;j++){
//去掉自身
if(i==t.x&&j==t.y) continue;
//判断是否出界
if(i<0||i>=n||j<0||j>=n) continue;
//当遍历到的点的高度和当前位置高度不一致时 此时需要进行判断比当前低or高
if(g[i][j]!=g[t.x][t.y]){
if(g[i][j]>g[t.x][t.y]) has_higher = true;
else has_lower = true;
}
//连通块中一定是高度相同的点构成的 且满足之前没有被访问过
else if(!st[i][j]){
q[++tt] = {i,j};
st[i][j] = true;
}
}
}
}
int main(){
cin >> n;
//读入部分
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
cin>>g[i][j];
//定义答案 peak为山峰个数 valley为山谷个数
int peak = 0,valley = 0;
//Flood Fill算法模板 循环遍历+条件判断+统计
for(int i=0;i<n;i++)
for(int j=0;j<n;j++){
//当该点没有被访问过时
if(!st[i][j]){
//定义两个bool类型的变量 用引用的方式传入bfs函数
//has_higher表示是否有比当前位置高的地点
//has_lower表示是否有比当前位置低的地点
bool has_higher = false;
bool has_lower = false;
bfs(i,j,has_higher,has_lower);
//如果没有比当前位置高的地点 peak+1
if(!has_higher) peak++;
//同理 这里可能会出现全部都一样高度 所以两次统计都是if互不影响
if(!has_lower) valley++;
}
}
cout<<peak<<" ";
cout<<valley<<endl;
return 0;
}