BZOJ1102: [POI2007]山峰和山谷Grz

联通：八联通；山峰：一相同数字联通块周围数字都比他们小；山谷：一相同数字联通块周围数字都比他们大。问山峰山谷数。

搜。

#include<stdio.h>
#include<string.h>
//#include<queue>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
using namespace std;

#define LL long long
int qread()
{
    char c; int s=0,t=1; while ((c=getchar())<'0' || c>'9') (c=='-') && (t=-1);
    do s=s*10+c-'0'; while ((c=getchar())>='0' && c<='9'); return s*t;
}

//Pay attention to LL and double of qread!!!!

int n;
#define maxn 1011
int mp[maxn][maxn];

int qx[maxn*maxn],qy[maxn*maxn],head,tail;
bool vis[maxn][maxn];
int A=0,B=0;
const int dx[]={0,0,1,1,1,-1,-1,-1},dy[]={1,-1,1,0,-1,1,0,-1};
bool cf(int x,int y)
{
    for (int i=0;i<8;i++)
    {
        int nx=x+dx[i],ny=y+dy[i];
        if (nx<1 || nx>n || ny<1 || ny>n || mp[nx][ny]==mp[x][y]) continue;
        if (mp[nx][ny]>mp[x][y]) return 0;
    }
    return 1;
}
bool cg(int x,int y)
{
    for (int i=0;i<8;i++)
    {
        int nx=x+dx[i],ny=y+dy[i];
        if (nx<1 || nx>n || ny<1 || ny>n || mp[nx][ny]==mp[x][y]) continue;
        if (mp[nx][ny]<mp[x][y]) return 0;
    }
    return 1;
}
void bfs(int sx,int sy)
{
    bool f=1,g=1;
    head=0; tail=1; qx[0]=sx; qy[0]=sy; vis[sx][sy]=1;
    while (head!=tail)
    {
        int xx=qx[head],yy=qy[head]; head++;
        if (f) f=cf(xx,yy); if (g) g=cg(xx,yy);
        for (int i=0;i<8;i++)
        {
            int nx=xx+dx[i],ny=yy+dy[i];
            if (nx<1 || nx>n || ny<1 || ny>n || vis[nx][ny] || mp[nx][ny]!=mp[xx][yy]) continue;
            vis[nx][ny]=1; qx[tail]=nx; qy[tail]=ny; tail++;
        }
    }
    if (f) A++; if (g) B++;
}

int main()
{
    n=qread();
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            mp[i][j]=qread();
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            if (!vis[i][j]) bfs(i,j);
    printf("%d %d
",A,B);
    return 0;
}