【雙向BFS模板】

文章目錄

原題鏈接:

https://ac.nowcoder.com/acm/contest/549/G
是個裸的雙向BFS的題,看到一份比較好的代碼,我以前好像都是寫的兩個BFS的,而他這個寫在一起了,而且更好得體現了一層一層的搜索的思想
比如原來做題的時候就遇到過,因爲我是 輸出調試法來debug的,所以原來打印座標的時候不知道這個點是在第幾步搜出來的,其實搜完一層後,新加的就是下一層了
這個就作爲一個模板了

寫的時候還寫錯了,第二個找到目標後馬上就退出,所以BFS裏面的return要寫在循環外面

#include"bits/stdc++.h"
#define out(x) cout<<#x<<"="<<x
#define C(n,m) (m>n?0:(long long)fac[(n)]*invf[(m)]%MOD*invf[(n)-(m)]%MOD)
using namespace std;
typedef long long LL;
const int maxn=1e3+5;
const int MOD=1e9+7;
char G[maxn][maxn];
bool vis[2][maxn][maxn];
int N,M;
struct Node
{
	int x,y;
	Node() {}
	Node(int x,int y):x(x),y(y) {}
};
Node ST1,ST2;//表示兩個起點
int dx[8]= {1,0,-1,0,1,1,-1,-1};
int dy[8]= {0,1,0,-1,1,-1,1,-1};
queue<Node>que[2];
int BFS(int cmd)
{
	int sz=que[cmd].size();//精髓~隊列裏有的就是上一次各個方向搜索的
	while(sz--)
	{
		Node u=que[cmd].front();
		que[cmd].pop();
		for(int i=0; i<4+(cmd==0?4:0); i++) //第一個是8個方向
		{
			int xx=u.x+dx[i];
			int yy=u.y+dy[i];
			if(xx<1||xx>N||yy<1||yy>M||G[xx][yy]=='#'||vis[cmd][xx][yy])continue;
			if(vis[cmd^1][xx][yy])return 1;//要在這裏一找到就退出,不能像普通的寫在循環外面 
			vis[cmd][xx][yy]=1;
			que[cmd].push(Node(xx,yy));
		}
	}
	return 0;
}
int solve()
{
	while(!que[0].empty())que[0].pop();
	while(!que[1].empty())que[1].pop();
	que[0].push(ST1);
	que[1].push(ST2);
	memset(vis,0,sizeof vis);
	vis[0][ST1.x][ST1.y]=1;
	vis[1][ST2.x][ST2.y]=1;
	int step=0;
	while(que[0].size()||que[1].size())//只要有一個還能走就繼續走
	{
		step++;
		if(BFS(0))return step;
		if(BFS(1))return step;
		if(BFS(1))return step;//第二個人要走兩步,所以這裏弄兩次
	}
	return -1;
}
int main()
{

	while(cin>>N>>M)
	{
		for(int i=1; i<=N; i++)
		{
			for(int j=1; j<=M; j++)
			{
				cin>>G[i][j];
				if(G[i][j]=='C')ST1=Node(i,j);
				else if(G[i][j]=='D')ST2=Node(i,j);
			}
		}
		int ans=solve();
		if(ans==-1)puts("NO");
		else
		{
			puts("YES");
			cout<<ans<<endl;
		}

	}
}

/*
4 5
. C # # . 
# . . # . 
# # . D . 
# . # . . 

*/
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章