BZOJ 1102 [POI2007]山峰和山谷Grz

题目链接

https://www.lydsy.com/JudgeOnline/problem.php?id=1102

思路

DFS（貌似会爆栈？BZOJ A不了，我们学校OJ和洛谷都 A了）

递归搜索，如果一块围起来的地方符合山峰/山谷的定义，那么记录答案，注意这一种情况：

2 -5 -3
-1 2 -1
-1 -1 2

这样一种样例，如果从左上角开始搜，RP不好可能把(1,1),(2,2)判掉了，但是(3,3)没有搜索，由于我是搜到一个地方就把那个地方打上标记（保证时间复杂度），搜索(3,3)就会出事，需要特判一下这种情况。具体参考代码。

代码

#include <cstdio>
#include <cstring>

const int maxn=1000;
const int dx[]={1,1,1,0,0,-1,-1,-1};
const int dy[]={1,0,-1,1,-1,1,0,-1};

int n,h[maxn+10][maxn+10],vis[maxn+10][maxn+10],hi,lo;

int search_lo(int x,int y)
{
  vis[x][y]=1;
  int flag=1;
  for(register int i=0; i<8; ++i)
    {
      int nx=x+dx[i],ny=y+dy[i];
      if((nx<=0)||(nx>n)||(ny<=0)||(ny>n))
        {
          continue;
        }
      if(h[nx][ny]<h[x][y])
        {
          flag=0;
          break;
        }
      if((h[nx][ny]==h[x][y])&&(vis[nx][ny]==-1))
        {
          flag=0;
          break;
        }
    }
  if(!flag)
    {
      vis[x][y]=-1;
      return 0;
    }
  for(register int i=0; i<8; ++i)
    {
      int nx=x+dx[i],ny=y+dy[i];
      if((nx<=0)||(nx>n)||(ny<=0)||(ny>n))
        {
          continue;
        }
      if((h[nx][ny]==h[x][y])&&(!vis[nx][ny]))
        {
          if(!search_lo(nx,ny))
            {
              vis[x][y]=-1;
              return 0;
            }
        }
    }
  return 1;
}

int search_hi(int x,int y)
{
  vis[x][y]=1;
  int flag=1;
  for(register int i=0; i<8; ++i)
    {
      int nx=x+dx[i],ny=y+dy[i];
      if((nx<=0)||(nx>n)||(ny<=0)||(ny>n))
        {
          continue;
        }
      if(h[nx][ny]>h[x][y])
        {
          flag=0;
          break;
        }
      if((h[nx][ny]==h[x][y])&&(vis[nx][ny]==-1))
        {
          flag=0;
          break;
        }
    }
  if(!flag)
    {
      vis[x][y]=-1;
      return 0;
    }
  for(register int i=0; i<8; ++i)
    {
      int nx=x+dx[i],ny=y+dy[i];
      if((nx<=0)||(nx>n)||(ny<=0)||(ny>n))
        {
          continue;
        }
      if((h[nx][ny]==h[x][y])&&(!vis[nx][ny]))
        {
          if(!search_hi(nx,ny))
            {
              vis[x][y]=-1;
              return 0;
            }
        }
    }
  return 1;
}

int main()
{
  scanf("%d",&n);
  for(register int i=1; i<=n; ++i)
    {
      for(register int j=1; j<=n; ++j)
        {
          scanf("%d",&h[i][j]);
        }
    }
  for(register int i=1; i<=n; ++i)
    {
      for(register int j=1; j<=n; ++j)
        {
          if(!vis[i][j])
            {
              lo+=search_lo(i,j);
            }
        }
    }
  memset(vis,0,sizeof vis);
  for(register int i=1; i<=n; ++i)
    {
      for(register int j=1; j<=n; ++j)
        {
          if(!vis[i][j])
            {
              hi+=search_hi(i,j);
            }
        }
    }
  printf("%d %d\n",hi,lo);
  return 0;
}