山峰山谷bfs

题目描述】

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

定义山峰和山谷如下：

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

所有方格高度都相同；

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

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

【输入】

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

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

【输出】

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

【输入样例】

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

【提示】

样例1解释：

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

样例输出2:3 3

样例2解释：

思路

直接广搜就可以，深搜的话深度太大，当前点坐标为x,y;遍历周围的八个点，记录周围点的大小情况，如果不一致就标记。

#include <iostream>
#include <cstdio>
using namespace std;
typedef long long ll;
int book[1001][1001];//标记是否访问过
ll arr[1001][1001];
struct f {
    int x;
    int y;
} brr[1000005];
int nextx[]= {-1,-1,0,1,1,1,0,-1};
int nexty[]= {0,1,1,1,0,-1,-1,-1};
int m,k;
int sum1 = 0 ,sum2=0;
void bfs(int x, int y)
{
    int head = 1, tail = 1;
    int i;
    int flag = 0;//标记是否为山峰或者山丘
    brr[head].x = x;
    brr[head].y = y;
    tail++;
    while (head < tail)
    {
        for (i = 0; i < 8; i++)
        {
            int nx = brr[head].x + nextx[i];
            int ny = brr[head].y + nexty[i];
            if (nx<0 || nx>m - 1 || ny<0 || ny>m - 1)
                continue;
            if (arr[nx][ny] != arr[x][y])//如果不等于当前探索的点，判断大小情况
            {
                if (k == 1)
                {
                    if (arr[nx][ny] < arr[x][y])
                    {//当前点与上一个点相对于探索点的大小不一致，标记一下此探索点构不成山峰或山丘
                        flag = 1;
                       // break;不能直接退出，要遍历完探索点组成的连通块后退出循环
                    }
                }
                else if (k == 2)
                {
                    if (arr[nx][ny] > arr[x][y])
                    {
                        flag = 1;
                       // break;
                    }
                }
                else {//探索一个山丘或山峰的时候，将第i=0时的点的大小标记一下
                    if (arr[nx][ny] > arr[x][y])k = 1;
                    if (arr[nx][ny] < arr[x][y])k = 2;
                }
            }
            else if((book[nx][ny]==0)&&(arr[nx][ny]==arr[x][y]))
            {//如果与探索点的大小一致，加入队列继续探索
                book[nx][ny] = 1;
                brr[tail].x = nx;
                brr[tail].y = ny;
                tail++;
            }
        }
        head++;
    }
    if (flag == 0)
    {//表明（x,y）点组成的连通块可以成为山峰或山丘
        if (k == 0)
        {//整个地图方格的高度均相同，则整个地图既是一个山谷，也是一个山峰。
            sum1++;
            sum2++;
        }
        else {
            if (k == 1) {sum1++;
            }
            else if (k == 2) { sum2++; }
        }
    }
}
int main(void)
{
    cin >> m;
    for (int i = 0; i < m; i++)
    {
        for (int j = 0;j < m; j++)
        {
            scanf_s("%lld", &arr[i][j]);
        }
    }
    for (int i = 0; i < m; i++)
    {
        for (int j = 0; j < m; j++)
        {//探索（i,j）点组成的连通块是否可以成为一个山峰或山丘，点（i,j）称为探索点
            if (book[i][j] == 0)
            {
                k = 0;
                book[i][j] = 1;
                bfs(i,j);
            }
        }
    }
    cout << sum2 << ' ' << sum1;
    return 0;
}