山峰和山谷（信息学奥赛一本通-T1454）

【题目描述】

给定一个 n×n 的网格状地图，每个方格 (i,j)有一个高度 wij​​ 。如果两个方格有公共顶点，则它们是相邻的。

定义山峰和山谷如下：

均由地图上的一个连通块组成；

所有方格高度都相同；

周围的方格（即不属于山峰或山谷但与山峰或山谷相邻的格子）高度均大于山谷的高度，或小于山峰的高度。

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

【输入】

第一行一个整数n（2≤n≤1000），表示地图的大小。

接下来 n 行每行 n 个整数表示地图。第 i 行有 n 个整数 wi1,wi2,…,win(0≤wij≤1000000000)，表示地图第 i 行格子的高度。

【输出】

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

【输入样例】

输入样例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

输入样例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

【输出样例】

输出样例1：

2 1

样例解释：

输出样例2：

3 3

样例解释：

思路：基础的 BFS，只要按照高度排序后，分别 BFS 山峰、山谷即可

【源程序】

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<bitset>
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
#define Pair pair<int,int>
LL quickPow(LL a,LL b){ LL res=1; while(b){if(b&1)res*=a; a*=a; b>>=1;} return res; }
LL multMod(LL a,LL b,LL mod){ a%=mod; b%=mod; LL res=0; while(b){if(b&1)res=(res+a)%mod; a=(a<<=1)%mod; b>>=1; } return res%mod;}
LL quickMultPowMod(LL a, LL b,LL mod){ LL res=1,k=a; while(b){if((b&1))res=multMod(res,k,mod)%mod; k=multMod(k,k,mod)%mod; b>>=1;} return res%mod;}
LL quickPowMod(LL a,LL b,LL mod){ LL res=1; while(b){if(b&1)res=(a*res)%mod; a=(a*a)%mod; b>>=1; } return res; }
LL getInv(LL a,LL mod){ return quickPowMod(a,mod-2,mod); }
LL GCD(LL x,LL y){ return !y?x:GCD(y,x%y); }
LL LCM(LL x,LL y){ return x/GCD(x,y)*y; }
const double EPS = 1E-6;
const int MOD = 1000000000+7;
const int N = 1000+5;
const int dx[] = {0,0,1,-1,1,1,-1,-1};
const int dy[] = {1,-1,0,0,1,-1,1,-1};
using namespace std;
struct Node {
    int x, y;
    int val;
    Node() {}
    Node(int x, int y) : x(x), y(y) {}
    Node(int x, int y, int val) : x(x), y(y), val(val) {}
    bool operator<(const Node &rhs) const { return val < rhs.val; }
} node[N * N];
int n;
int G[N][N];
bool vis[N][N];
void downBFS(int x, int y) {
    queue<Node> Q;
    Q.push(Node(x, y));
    vis[x][y] = true;
    while (!Q.empty()) {
        Node now = Q.front();
        Q.pop();
        int x = now.x, y = now.y;
        for (int i = 0; i < 8; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx <= 0 || ny <= 0 || nx > n || ny > n)
                continue;
            if (!vis[nx][ny] && G[x][y] <= G[nx][ny]) {
                vis[nx][ny] = true;
                Q.push(Node(nx, ny));
            }
        }
    }
}
void upBFS(int x, int y) {
    queue<Node> Q;
    Q.push(Node(x, y));
    vis[x][y] = true;
    while (!Q.empty()) {
        Node now = Q.front();
        Q.pop();
        int x = now.x, y = now.y;
        for (int i = 0; i < 8; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx <= 0 || ny <= 0 || nx > n || ny > n)
                continue;
            if (!vis[nx][ny] && G[x][y] >= G[nx][ny]) {
                vis[nx][ny] = true;
                Q.push(Node(nx, ny));
            }
        }
    }
}
int main() {
    while (scanf("%d", &n) != EOF) {
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                scanf("%d", &G[i][j]);
                int pos = (i - 1) * n + j;
                node[pos] = Node(i, j, G[i][j]);
            }
        }
        sort(node + 1, node + 1 + n * n);

        int down = 0;
        memset(vis, false, sizeof(vis));
        for (int i = 1; i <= n * n; i++) {
            int x = node[i].x;
            int y = node[i].y;
            if (!vis[x][y]) {
                downBFS(x, y);
                down++;
            }
        }
        int up = 0;
        memset(vis, false, sizeof(vis));
        for (int i = n * n; i >= 1; i--) {
            int x = node[i].x;
            int y = node[i].y;
            if (!vis[x][y]) {
                upBFS(x, y);
                up++;
            }
        }
        printf("%d %d\n", up, down);
    }
    return 0;
}