Djkstra使用邻接表+优先队列优化
#include <cstdio>
#include <vector>
#include <algorithm>
#include <queue>
#include <set>
const int MAXN = 105;
using namespace std;
typedef pair<int, int> V;
int n,INF = 1000000000;
vector<V> G[MAXN];
int d[MAXN],visited[MAXN]={0};
vector<int> pre[MAXN];
void Djkstra(int start)
{
int i,j,k;
priority_queue<V, vector<V>, greater<V> > Q;
fill(d,d+MAXN,INF);
Q.push(V(0,start));
d[start] = 0;
for(i=0;i<n;i++){
int u = -1;
while (!Q.empty()){
if(!visited[Q.top().second]){
u = Q.top().second;
Q.pop();
break;
}
Q.pop();
}
if(u==-1)
return;
visited[u] = 1;
for(j=0;j<G[u].size();j++){
int v = G[u][j].second, duv = G[u][j].first;
if(!visited[v] && d[v] > d[u]+duv){
d[v] = d[u]+duv;
pre[v].clear();
pre[v].push_back(u);
Q.push(V(d[v], v));
}else if(!visited[v] && d[v] == d[u]+duv){
pre[v].push_back(u);
}
}
}
}
注释:G是邻接表,pair<int, int> V第一个是边的权值,第二个是终止节点v。u,v是一条边的起始、终止节点。
visited是这个节点是否已经被纳入了正在扩张的集合,d是出发节点到其他节点的最小距离,pre是一个节点的前驱节点集合,搞出来之后可以结合DFS求题目要求的一系列东西。
这里的优先队列priority_queue<V, vector, greater > Q不用写inq, 写了反而会错