POJ 3767 I Wanna Go Home G++ dijkstra实现 spfa未实现 背

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
//英语 看博友分析 抄博友程序   dijkstra实现  spfa未实现   背
//算导dijkstra用优先队列 百度百科的dijkstra没用优先队列 
int N,M;
int mp[608][608]; 
int dui[608];
int dis1[608];
int dis2[608];
int vis[608];
int l[608];
int r[608];
void dij(int s,int dis[])//背 
{
	for(int i=1;i<=N;i++)
	{
		dis[i]=mp[s][i];//背 
	}
	vis[1]=1; 
	for(int i=1;i<=N;i++)
	{
		int xiao=-1;
		int wz=-1;
		for(int j=1;j<=N;j++)
		{
			if((xiao==-1 || xiao>dis[j])&&vis[j]==0&&dui[j]==dui[s]&&dis[j]!=-1)//背 
			{
				xiao=dis[j];
				wz=j; 
			}
		}
		if(xiao==-1)//抄博友程序 
		{
			break;
		}
		vis[wz]=1;
		for(int j=1;j<=N;j++)
		{
			if(vis[j]==0 && dui[j]==dui[s] && mp[wz][j]!=-1 &&(dis[j]==-1 ||dis[j]>dis[wz]+mp[wz][j]))
			{
				dis[j]=dis[wz]+mp[wz][j];
			}
		} 
	}
}
int main()
{
	while(1)
	{
		scanf("%d",&N);
		if(N==0)
		{
			break;
		}
		scanf("%d",&M);
		for(int i=1;i<=N;i++)
		{
			for(int j=1;j<=N;j++)
			{
				mp[i][j]=-1;
				if(i==j)
				{
					mp[i][j]=0;
				}
			}
		} 
		memset(dui,0,sizeof(dui));
		memset(dis1,-1,sizeof(dis1));
		memset(dis2,-1,sizeof(dis2));
		//cout<<"hi"<<endl;
		for(int i=1;i<=M;i++)
		{
			int x,y,zhi;
			scanf("%d%d%d",&x,&y,&zhi);
			if(mp[x][y]==-1||mp[x][y]>zhi)
			mp[x][y]=mp[y][x]=zhi;
		}
		int len=0;
		int ren=0;
		for(int i=1;i<=N;i++)
		{
			int t;
			scanf("%d",&t);
			dui[i]=t;
			if(dui[i]==1)//抄博友程序 
			{
				l[len++]=i;
			}else
			{
				r[ren++]=i;
			}
		}
		memset(vis,0,sizeof(vis));
		dij(1,dis1);
		memset(vis,0,sizeof(vis));
		dij(2,dis2);
		int jg=-1;
		for(int i=0;i<len;i++)
		{
			for(int j=0;j<ren;j++)
			{
				if(dis1[l[i]]!=-1&& dis2[r[j]]!=-1 && mp[l[i]][r[j]]!=-1)
				if(jg==-1||jg>dis1[l[i]]+dis2[r[j]]+mp[l[i]][r[j]])
				{
					jg=dis1[l[i]]+dis2[r[j]]+mp[l[i]][r[j]];
				} 
			} 
		}
		if(jg==-1)
		{
			printf("-1\n");
		}else
		{
			printf("%d\n",jg);
		}
	} 
	return 0;
}