ACM: HDU 1874 畅通工程续-Dijkstra算法

HDU 1874 畅通工程续

Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

Description

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

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

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

/*/
Dijkstra[简称Dij]算法跟BFS很类似，或者可以说，BFS就是一种特殊的Dij。

Dij的核心就只有三行。然后用到优先队列。

下面还介绍了pair对象的使用方法；

AC代码：
/*/

#include"algorithm"
#include"iostream"
#include"cstring"
#include"cstdlib"
#include"string"
#include"cstdio"
#include"vector"
#include"cmath"
#include"queue"
using namespace std;
typedef long long LL;
#define memset(x,y) memset(x,y,sizeof(x))
#define memcpy(x,y) memcpy(x,y,sizeof(x))
#define MX 401

/*/************************************************************* /*/
//     	typedef pair< T, int> PII;   //  pair<int,int>
/*/*************************************************************
pair 默认对first升序，当first相同时对second升序；

类模板：template <class T1, class T2> struct pair

参数：T1是第一个值的数据类型，T2是第二个值的数据类型。

功能：pair将一对值组合成一个值，这一对值可以具有不同的数据类型（T1和T2）
两个值可以分别用pair的两个公有函数first和second访问。

************************************************************* /*/

const int dij_v=1e3;
const int dij_edge=1e4;

template<class T> //模版类。
struct Dijkstra {

	typedef pair< T, int> PII;   //  pair<int,int>
	struct Edge {
		int v,nxt;
		T w;
	} E[dij_edge<<1];
	int Head[dij_v],erear;
	T dis[dij_v],INF;

	void edge_init() {
		memset(Head,-1);
		erear=0;
	}

	void edge_add(int u,int v,T w) {
		E[erear].v=v;
		E[erear].w=w;
		E[erear].nxt=Head[u];
		Head[u]=erear++;
	}

	void run(int u) {
		memset(dis,0x3f);
		INF=dis[0];
		priority_queue<PII, vector<PII>, greater<PII> >Q;  //优先队列
		while(!Q.empty())Q.pop();
		Q.push(PII(0,u));
		dis[u]=0;
		while(!Q.empty()) {
			PII A=Q.top();
			Q.pop();
			int a=A.second;
			if(A.first!=dis[a])continue;  //如果不在一棵最小生成树上，跳过。
			for(int i=Head[a]; ~i; i=E[i].nxt) {
				int v=E[i].v;
				T w=E[i].w;

				/*/********************Dijkstra算法核心************************* /*/
				if(w+dis[a]<dis[v]) {    // 判断是否是到达改点的最小路
					dis[v]=dis[a]+w;	 // 如果不是就覆盖掉改点
					Q.push(PII(dis[v],v));	 // 记录进队列。
				}
				/*/************************************************************* /*/

			}
		}
	}
};

Dijkstra<int > dij;

int main() {
	int n,m;
	while(~scanf("%d%d",&n,&m)) {
		dij.edge_init();
		for(int i=1; i<=m; i++) {
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			dij.edge_add(u,v,w);
			dij.edge_add(v,u,w);
		}
		int st,ed;
		scanf("%d%d",&st,&ed);
		dij.run(st);
		if(dij.dis[ed]==dij.INF)puts("-1");
		else printf("%d\n",dij.dis[ed]);
	}
	return 0;
}