如此,M条路中凡是连接属于不同集合的城市的路为单向路,只能从集合1中的城市走向集合2, 在此基础上用Dijkstra算法求最短路即可
#include
#include
#define INF 1000000
int map[606][606];
int visit[601];
int dist[601];
int leap[601];
int n,m,p;
void dij()
{
int i,j,min,flag;
for(i=1;imap[p][j]+dist[p]))
dist[j]=map[p][j]+dist[p];
if(!visit[j]&&min>dist[j])
{
min=dist[j];
flag=j;
}
}
p=flag;
}
if(dist[2]==INF)
printf("%d\n",-1);
else
printf("%d\n",dist[2]);
}
int main()
{
int i,j,x,y,val;
while(scanf("%d",&n),n)
{
scanf("%d",&m);
for(i=1;i<=n;i++)
{
visit[i]=0;
for(j=1;j<=i;j++)
map[i][j]=map[j][i]=INF;
}
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&val);
map[x][y]=map[y][x]=val;
}
for(i=1;i<=n;i++)
scanf("%d",&leap[i]);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
{
if(leap[i]!=leap[j])
{
if(leap[i]==1)
map[j][i]=INF;
else
map[i][j]=INF;
}
}
for(i=1;i<=n;i++)
dist[i]=map[1][i];
dist[1]=0;
p=1;
dij();
}
return 0;
}