山峰和山谷

原题链接

题目描述

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

求地图内山峰和山谷的数量。特别地，如果整个地图方格的高度均相同，则整个地图既是一个山谷，也是一个山峰。

【输入】
第一行一个整数n（2≤n≤1000），表示地图的大小。
接下来 n 行每行 n 个整数表示地图。第 i 行有 n 个整数 $wi1,wi2,\dots ,win(0\le wij\le 1000000000)$，表示地图第 i 行格子的高度。

【输出】
输出一行两个整数，分别表示山峰和山谷的数量。

样例

in:
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

out:
2 1

---

算法

Flood Fill
对于周围全部的山的判断，通过传引用bool来标记，或开全局
bfs搜索，防止dfs爆栈。

时间复杂度

$O(n^2)$

c++代码

#include <bits/stdc++.h>
using namespace std;
typedef long long L;
const int N = 1e3 + 5;
typedef pair<int,int> P;
int n, w[N][N];
bool v[N][N];
P q[N*N];
void  bfs(int x, int y, bool& d, bool& f) {
    int tt = -1, hh = 0;
    q[++tt] = { x,y };
    while (hh <= tt) {
        P t = q[hh++];
        x = t.first, y = t.second;
        for (int i = x - 1; i <= x + 1; ++i) {
            for (int j = y - 1; j <= y + 1; ++j) {
                if (i<1 ||i>n||j<1||j>n) continue;
                if (w[x][y] != w[i][j]) {
                    if (w[i][j] > w[x][y]) d = 1;
                    else f = 1;

                }
               else if(!v[i][j]) {//为什么v[][]在后面判，因为在判w[x][y]与w[i][j]是不论有没有搜过，只是比周围大小。
                    v[i][j] = 1;
                    q[++tt] = { i,j };
                }
            }
        }
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            cin >> w[i][j];
        }
    }
       int  down = 0, top = 0;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            if (!v[i][j]) {
                v[i][j] = 1;
                bool d = 0, f = 0;
                bfs(i, j, d, f);
                if (!d) top++;
                if (!f) down++;
            }
        }
    }
    cout << top << ' ' << down;
}
 

---