信息学奥赛一本通 山峰和山谷

地址 http://ybt.ssoier.cn:8088/problem_show.php?pid=1454

BFS

#include <iostream>
#include <queue>

using namespace std;

/*
给定一个 n×n 的网格状地图，每个方格 (i,j)有一个高度 wij​​ 。如果两个方格有公共顶点，则它们是相邻的。
定义山峰和山谷如下：
均由地图上的一个连通块组成；
所有方格高度都相同；
周围的方格（即不属于山峰或山谷但与山峰或山谷相邻的格子）高度均大于山谷的高度，或小于山峰的高度。
求地图内山峰和山谷的数量。特别地，如果整个地图方格的高度均相同，则整个地图既是一个山谷，也是一个山峰。

【输入】
第一行一个整数n（2≤n≤1000），表示地图的大小。
接下来 n 行每行 n 个整数表示地图。第 i 行有 n 个整数 wi1,wi2,…,win(0≤wij≤1000000000)，表示地图第 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
*/


const int N = 1010;
int arr[N][N];
int vis[N][N];
int n;

int addX[8] = { 1,-1,0,0,1,-1,1,-1};
int addY[8] = { 0,0,1,-1,1,-1,-1,1};

void bfs(int x, int y, int& hasLower, int& hasHigher) {
vis[x][y] = 1;

queue<pair<int, int> > q;
pair<int, int> pa = make_pair(x, y);
q.push(pa);

while (!q.empty()) {
int currx = q.front().first;
int curry = q.front().second;
q.pop();

for (int i = 0; i < 8; i++) {
int newx = currx + addX[i];
int newy = curry + addY[i];

if (newx >= 0 && newx < n && newy >= 0 && newy < n) {
if (arr[newx][newy] > arr[x][y]) {
hasHigher++;
}
else if (arr[newx][newy] < arr[x][y]) {
hasLower++;
}
else if (arr[newx][newy] == arr[x][y] && vis[newx][newy] == 0) {
pair<int, int> pa = make_pair(newx, newy);
q.push( pa ); vis[newx][newy] = 1;
}

}
}
}
}


int main()
{
cin >> n;

for (int x = 0; x < n; x++) {
for (int y = 0; y < n; y++) {
cin >> arr[x][y];
}
}
int peak = 0; int valley = 0; //山峰山谷计数
for (int x = 0; x < n; x++) {
for (int y = 0; y < n; y++) {
int hasLower = 0; int hasHigher = 0;
if(vis[x][y] == 0){
bfs(x,y, hasLower, hasHigher);
if (hasLower == 0 && hasHigher == 0) {peak++, valley++;}
else if (hasLower == 0 && hasHigher != 0) { valley++; }
else if (hasLower != 0 && hasHigher == 0) { peak++; }
}
}
}

cout << peak << " " << valley << endl;

return 0;
}