山峰和山谷
题目描述
给定一个n*n的网格状地图,每个方格(i,j)有一个高度
w
i
,
j
w_i,_j
wi,j。如果两个方格有公共顶点,则它们是相邻的。
定义山峰山谷如下:
均由地图上的一个联通块组成。
所有方格高度都相同。
周围的方格(即不属于山峰或山谷但与山峰或山谷相邻的格子)高度均大于山谷的高度,或小于山峰的高度。
求地图内山峰和山谷的数量。特别的,如果整个地图方格的高度均相同,则整个地图即是一个山谷,也是一个山峰。
输入格式
第一行一个正整数n,表示地图的大小。
接下来n行,每行n个整数表示地图。第i行有n个正整数
w
i
,
1
,
w
i
,
2
,
w
i
,
3
,
.
.
.
,
w
i
,
n
w_i,_1,w_i,_2,w_i,_3,...,w_i,_n
wi,1,wi,2,wi,3,...,wi,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;
}