1106. 山峰和山谷变种的搜索 求连通块

Posted 2021-11-04 幽殇默

https://www.acwing.com/problem/content/description/1108/
本质上是一个特别裸的求联通块的问题，但是有一点的变种，就是我们要判断周围的数是不是都比它大，
周围的数是不是都比它小，所以可以用两个bool变量判断。
如果没有一个比它高的则必为山峰。
如果没有一个比它第低的则必为山谷。

#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int a[N][N],n,cnt1,cnt2;
bool st[N][N];
int dx[10]={-1,-1,-1,0,0,1,1,1};
int dy[10]={-1,0,1,-1,1,-1,0,1};
void bfs(int x,int y,bool &flag1,bool &flag2)
{
    st[x][y]=1;
    queue<pair<int,int>>q;  q.push({x,y});
    while(q.size())
    {
        auto u=q.front(); q.pop();
        x=u.first,y=u.second;
        for(int i=0;i<8;i++)
        {
            int tempx=x+dx[i],tempy=y+dy[i];
            if(tempx<0||tempx>=n||tempy<0||tempy>=n) continue;
            if(a[x][y]>a[tempx][tempy]) flag1=1;
            else if(a[x][y]<a[tempx][tempy]) flag2=1;
            else if(!st[tempx][tempy]) st[tempx][tempy]=1,q.push({tempx,tempy});
        }
    }
}
int main(void)
{
    cin>>n;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            cin>>a[i][j];
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            if(!st[i][j]) 
            {
                bool flag1=false,flag2=false;
                bfs(i,j,flag1,flag2);
                if(!flag2) cnt1++;
                if(!flag1) cnt2++;
            }
    cout<<cnt1<<" "<<cnt2;
    return 0;
}