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≤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:
解析:
遍历每个点,但需要判断它周围的是不是比他高,或者比他矮,因为我要确定他是不是山峰和山谷
#include<bits/stdc++.h>
using namespace std;
const int N=1005;
#define x first
#define y second
int a[N][N];
int n,m;
typedef pair<int,int> PII;
bool st[N][N];
int cnt1,cnt2;
int dx[8]={0,0,-1,1,-1,1,-1,1};
int dy[8]={1,-1,0,0,1,1,-1,-1};
void bfs(int x,int y,bool &h,bool &l)
{
queue<PII> q;
q.push({x,y});
st[x][y]=true;
while(q.size())
{
auto t=q.front();
q.pop();
for(int i=0;i<8;i++)
{
int xx=t.x+dx[i];
int yy=t.y+dy[i];
if(xx<0||xx>=n||yy<0||yy>=n) continue;
if(a[t.x][t.y]!=a[xx][yy])
{
if(a[xx][yy]>a[t.x][t.y]) h=true; //山峰
else l=true;//山谷
}
else if(!st[xx][yy])
{
q.push({xx,yy});
st[xx][yy]=true;
}
}
}
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++) cin>>a[i][j];
memset(st,false,sizeof false);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
if(!st[i][j])
{
bool h=false,l=false;
bfs(i,j,h,l);
if(!h) cnt1++;
if(!l) cnt2++;
}
}
cout<<cnt1<<" "<<cnt2<<endl;
}