AcWing 1106. 山峰和山谷 bfs+思维

AcWing 1106. 山峰和山谷

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

这道题求的是山峰数和山谷数，我们发现山峰和山谷的限定条件是周围的都比他大，要么比他小，高度相等，所以我们在原有的bfs模板增加几个限定条件就可，我们在bfs的时候，我们把高度比它高的数量设为high，高度比它低的设为low,如果high或Low为0，则就增加山峰/山谷数。

代码如下：

#include<iostream>
#include<queue>
#define x first
#define y second
using namespace std;

typedef pair<int,int> PII;
const int N=1010;

int g[N][N];
bool st[N][N];
int dx[8]={0,1,0,-1,1,1,-1,-1};
int dy[8]={1,0,-1,0,1,-1,1,-1};
int n;
int res,cnt;

void bfs(int sx,int sy)
{
    st[sx][sy]=true;
    queue<PII> q;
    q.push({sx,sy});
    int h=g[sx][sy];
    int high=0,low=0;
    while(q.size())
    {
        auto t=q.front();
        q.pop();
        for(int i=0;i<8;i++)
        {
            int tx=t.x+dx[i];
            int ty=t.y+dy[i];
            if(tx<1||tx>n||ty<1||ty>n)
            continue;
            if(g[tx][ty]>h)
            high++;
            else if(g[tx][ty]<h)
            low++;
            else if(!st[tx][ty])
            {
                st[tx][ty]=true;
                q.push({tx,ty});
            }
        }

    }
    if(!high)
    res++;
    if(!low)
    cnt++;
}

int main(void)
{
    cin>>n;
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      cin>>g[i][j];
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
     if(!st[i][j])
     bfs(i,j);
    cout<<res<<" "<<cnt;
}