HDU 2112 使用优先队列优化Dijkstra算法

题目链接http://acm.hdu.edu.cn/showproblem.php?pid=2112
使用优先队列优化的Dijkstra算法，其时间复杂度为$O(ElogV)$。无论是稀疏图还是稠密图，优化后的运行速度都要比朴素Dijkstra快。
下面是几点需要了解的：

• 在Dijkstra算法中，dis[i]越小应该越先出队。在STL中我们可以使用greater<int>表示大于运算符，写成priority_queue<<int> , vector<int>, greater<int> > Q这种形式来声明一个小整数优先出队的队列。
• 代码第12行的pair表示将dis数组的值与编号进行捆绑，以后作为一个整体放到优先队列中。（需要说明的是pair类型的排序规则是首先比较第一维，相等时再比较第二维）。因此代码中第24行的priority_queue < P, vector<P>, greater<P> > Q;就定义了一个由二元组构成的优先队列。

AC代码：

#include <iostream>
#include <queue>
#include <vector>
#include <map>
#include <algorithm>  //fill函数的头文件
using namespace std;

//HDU	Accepted	2112	3151MS	2160K	1413 B	G++
const int MAX_N = 10010;
const int INF = 0x3f3f3f3f;
 
typedef pair<int, int> P;  //pair的第一维是当前最短距离，第二维是节点编号

struct edge {
	int to, cost;
	edge(int tt, int tc) : to(tt), cost(tc){};
};
vector<edge> G[MAX_N];  //邻接表
map<string, int> mp;
int n, cost;

int Dijkstra(int st, int ed) {
	int dis[MAX_N];
	fill(dis, dis+n+1, INF);
	//优先队列
	priority_queue < P, vector<P>, greater<P> > Q;
	dis[st] = 0;
	Q.push(P(0, st));  //把起始节点加入队列
	while(!Q.empty()) {
		P p = Q.top();
		Q.pop();
		int v = p.second;  //获取pair p的第二维，顶点的编号
		if(dis[v] < p.first)	continue;
		for(int i=0; i<G[v].size(); i++) {
			edge e = G[v][i];
			if(dis[e.to] > dis[v] + e.cost) {
				dis[e.to] = dis[v] + e.cost;
				Q.push(P(dis[e.to], e.to));
			}
		}
	}
	return dis[ed];
}

int main() {
	string st, ed, s, e;
	while(~scanf("%d", &n) && n != -1) {
		mp.clear();
		for(int i=0; i<MAX_N; i++)	G[i].clear();
		int cnt = 1;
		cin>>st>>ed;
		mp[st] = cnt++;
		mp[ed] = cnt++;

		for (int i=0; i<n; i++) {
			cin>>s>>e>>cost;
			if(!mp[s])	mp[s] = cnt++;
			if(!mp[e])	mp[e] = cnt++;
			//双向
			G[mp[s]].push_back(edge(mp[e], cost));
			G[mp[e]].push_back(edge(mp[s], cost));  //本来应该是e，结果开始的时候没注意打成了s，导致一直WA... 
		}
		int res = Dijkstra(mp[st], mp[ed]);
		if(res == INF)
			printf("-1\n");
		else
			printf("%d\n", res);
	}
	return 0;
}

---

此处再补充上P4779 【模板】单源最短路径（标准版）代码
这题如果直接用Dijsktra会TLE，注意初始化，少了邻接表的初始化是84分。下面是使用优先队列优化的100分代码：

#include <iostream>
#include <vector>
#include <queue>
#include <string.h>
using namespace std;
const int inf = 0x3f3f3f3f;

struct node {
	int to;
	int w;
};

int d[100005];
typedef pair<int, int> P;  //pair的第一维是当前最短距离，第二维是节点编号int d[100005];
vector<node> G[100005];

int main() {
	int n, m, s, u, v, w;
	cin>>n>>m>>s;  //s是起点 
	for(int i=0; i<=n; i++)	d[i] = inf, G[i].clear();
	for(int i=0; i<m; i++) {
		cin>>u>>v>>w;
		node e;
		e.to = v;
		e.w = w;
		G[u].push_back(e);
	}
	priority_queue<P, vector<P>, greater<P> > Q;
	while(!Q.empty())	Q.pop();
	Q.push(P(0, s));
	d[s] = 0; 
	while(!Q.empty()) {
		P p = Q.top();
		Q.pop();
		int v = p.second;  //节点编号 
		if(d[v] < p.first)	continue;
		for(int j=0; j<G[v].size(); j++) {
			node nd = G[v][j];
			if(d[nd.to] > d[v] + nd.w) {
				d[nd.to] = d[v] + nd.w;
				Q.push(P(d[nd.to], nd.to));
			}
		}
	}
	for(int i=1; i<n; i++)
		cout<<d[i]<<" ";
	cout<<d[n]<<endl;
	
	return 0;
}