POJ 1708 Game G++ 巧妙 bfs 没掌握

#include <iostream>
#include <cstdio>
#include <queue> 
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
//英语      看博友分析     抄博友程序       巧妙       bfs      没掌握 
int col[110];
int vis[110][110];
struct edge{
	int v;
	int c;
}; 
struct nod{
	int u;
	int v; 
	int d;//必须加 
};
queue<nod> que;
vector<edge> g[110];
int main()
{
	int N,L,K,Q;
	//while(cin>>N>>L>>K>>Q)
	while(scanf("%d%d%d%d",&N,&L,&K,&Q)!=EOF)
	{
		memset(vis,0,sizeof(vis));
		while(que.empty()!=1)
		{
			que.pop();
		}
		for(int i=0;i<110;i++)
		{
			g[i].clear();
		}
		for(int i=1;i<=N;i++)
		{
			scanf("%d",&col[i]);
		}
		int m;
		scanf("%d",&m);
		for(int i=0;i<m;i++)
		{
			int u,v,c;
			scanf("%d%d%d",&u,&v,&c);
			g[u].push_back((edge){v,c});
		}
		//int d=0;//wa
		int ans=20000;
		que.push((nod){L,K,0});
		vis[L][K]=1;
		while(que.empty()!=1)
		{
			int a=que.front().u;
			int b=que.front().v;
			int d=que.front().d;
			que.pop();
			//vis[a][b]=1;//bfs没掌握 
			if(a==Q ||b==Q)
			{
				//cout<<a<<" "<<b<<endl;
				ans=min(ans,d);
			}
			//d++;//wa 
			//cout<<a<<" "<<b<<" "<<d-1<<endl;			
			for(int i=0;i<g[a].size();i++)
			{
				int v=g[a][i].v;
				int c=g[a][i].c;
				if(c==col[b]&& vis[v][b]==0&& v!=b)
				{
					vis[v][b]=1;
					que.push((nod){v,b,d+1});
				}
			}
			for(int i=0;i<g[b].size();i++)
			{
				int v=g[b][i].v;
				int c=g[b][i].c;
				if(c==col[a] && vis[a][v]==0 && v!=a)
				{
					vis[a][v]=1;
					que.push((nod){a,v,d+1});
				}
			}
		}
		if(ans==20000)
		{
			cout<<"NO"<<endl;
		}else
		{
			cout<<"YES"<<endl;
			cout<<ans<<endl;
		}
	}
	return 0;
}