YbtOJ 广度搜索课堂过关 例2 山峰和山谷【bfs】

---

思路

这道题用广搜实现，我们可以bfs一次山峰，bfs一次山谷。
当一个范围周围所有数都没有它高时，它就是山峰；
当一个范围周围所有数都比它高时，它就是山谷。

$Code$

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int dx[9]={
    
    0,1,0,0,-1,1,1,-1,-1};
int dy[9]={
    
    0,0,1,-1,0,1,-1,-1,1};
int n,a[1010][1010],ans1,ans2;
int f[1000010][10],v[1010][1010],v1[1010][1010];
int bfs1(int x,int y)
{
    
    
	int hd=0,tl=1,w=1;
	f[1][1]=x;
	f[1][2]=y;
	v[x][y]=1;
	while(hd<tl)
	 {
    
    
	 	hd++;
	 	for(int i=1; i<=8; i++)
	 	 {
    
    
	 	 	int xx=f[hd][1]+dx[i];
	 	 	int yy=f[hd][2]+dy[i];
	 	 	if(a[xx][yy]>a[f[hd][1]][f[hd][2]]&&a[xx][yy]!=0)
	 	 	 	w=0;
	 	 	if(v[xx][yy]==0&&xx>=1&&xx<=n&&yy>=1&&yy<=n)
	 	 	 {
    
    
	 	 	 	if(a[xx][yy]==a[f[hd][1]][f[hd][2]])
	 	 	     {
    
    
	 	 	     	tl++;
	 	 	     	f[tl][1]=xx;
	 	 	     	f[tl][2]=yy;
	 	 	     	v[xx][yy]=1;
				 }
			 }
		 }
	 }
	return w;
}
int bfs2(int x,int y)
{
    
    
	int hd=0,tl=1,w=1;
	f[1][1]=x;
	f[1][2]=y;
	v1[x][y]=1;
	while(hd<=tl)
	 {
    
    
	 	hd++;
	 	for(int i=1; i<=8; i++)
	 	 {
    
    
	 	 	int xx=f[hd][1]+dx[i];
	 	 	int yy=f[hd][2]+dy[i];
	 	 	if(a[xx][yy]<a[f[hd][1]][f[hd][2]]&&a[xx][yy]!=0)
	 	 	 	w=0;
	 	 	if(v1[xx][yy]==0&&xx>=1&&xx<=n&&yy>=1&&yy<=n)
	 	 	 {
    
    
	 	 	 	if(a[xx][yy]==a[f[hd][1]][f[hd][2]])
	 	 	     {
    
    
	 	 	     	tl++;
	 	 	     	f[tl][1]=xx;
	 	 	     	f[tl][2]=yy;
	 	 	     	v1[xx][yy]=1;
				 }
			 }
		 }
	 }
	return w;
}
int main()
{
    
    
	cin>>n;
	for(int i=1; i<=n; i++)
	 for(int j=1; j<=n; j++)
	    cin>>a[i][j];
	for(int i=1; i<=n; i++)
	 for(int j=1; j<=n; j++)
	  {
    
    
	  	if(v[i][j]==0)
	  	 {
    
    
	  	 	int s1=bfs1(i,j);
	  	 	int s2=bfs2(i,j);
	  		//cout<<i<<' '<<j<<' '<<s1<<' '<<s2<<endl;
	  	 	ans1+=s1;
	  	 	ans2+=s2;
		 }
	  }
	cout<<ans1<<" "<<ans2;
	return 0;
}