L - 畅通工程续

点击打开链接

某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。 

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。

Input

本题目包含多组数据，请处理到文件结束。 
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000)，分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0～N-1编号。 
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。 
再接下一行有两个整数S,T(0<=S,T<N)，分别代表起点和终点。

Output

对于每组数据，请在一行里输出最短需要行走的距离。如果不存在从S到T的路线，就输出-1. 

Sample Input

3 3
0 1 1
0 2 3
1 2 1
0 2
3 1
0 1 1
1 2

Sample Output

2
-1

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
#define INF 0x3f3f3f3f
struct Pair
{
	int first,second;
	bool friend operator < (Pair a,Pair b)
	{
		return a.second > b.second;
	}
}pr,ne;
int n,m,s,t;
vector<int> edge[205];		 
int length[205][205];		
int dis[205];

void dijkstra()
{
	memset(dis,INF,sizeof(dis));
	bool vis[205];
	memset(vis,false,sizeof(vis));
	dis[s] = 0;
	pr.first = s;
	pr.second = 0;
	priority_queue<Pair> Q;
	Q.push(pr);
	while (!Q.empty())
	{
		pr = Q.top();
		Q.pop();
		
		if (vis[pr.first])
			continue;
		vis[pr.first] = true;
		
		for (int i = 0 ; i < edge[pr.first].size() ; i++)
		{
			ne.first = edge[pr.first][i];
			ne.second = pr.second + length[pr.first][ne.first];
			
			if (ne.second < dis[ne.first])
			{
				dis[ne.first] = ne.second;
				Q.push(ne);
			}
		}
	}
}

int main()
{
	while(~scanf ("%d%d",&n,&m))
	{
	  memset(edge,0,sizeof(edge));
	  memset(length,-1,sizeof(length));
	  for (int i = 1 ; i <= m ; i++)
	  {
		int x,y,z;
		scanf ("%d%d%d",&x,&y,&z);
		edge[x].push_back(y);
		edge[y].push_back(x);		
		
		if (length[x][y] == -1)
			length[x][y] = length[y][x] = z;
		else
			length[x][y] = length[y][x] = min(z,length[x][y]);
	  }
	  scanf("%d %d",&s,&t);
	  dijkstra();
	  printf ("%d\n",dis[t]==INF?-1:dis[t]);
    }
	  return 0;
}