HDU 1874 Dijkstra演算法 求任意兩個點之間的最短距離
阿新 • • 發佈:2018-11-10
題意:
某省自從實行了很多年的暢通工程計劃後,終於修建了很多路。不過路多了也不好,每次要從一個城鎮到另一個城鎮時,都有許多種道路方案可以選擇,而某些方案要比另一些方案行走的距離要短很多。這讓行人很困擾。
#include<cstdio> #include<cstring> #include<vector> #include<algorithm> #include<queue> using namespace std; #define INF 1e8 const int maxn = 200+10; int n,m; struct Edge { int from,to,dist; Edge(int f,int t,int d):from(f),to(t),dist(d){} }; struct HeapNode { int d,u; HeapNode(int d,int u):d(d),u(u){} bool operator< (const HeapNode &rhs)const { return d > rhs.d; } }; struct Dijkstra { int n,m; vector<Edge> edges; vector<int> G[maxn]; bool done[maxn]; int d[maxn]; void init(int n) { this->n=n; for(int i=0;i<n;i++) G[i].clear(); edges.clear(); } void AddEdge(int from,int to,int dist) { edges.push_back(Edge(from,to,dist)); m = edges.size(); G[from].push_back(m-1); } int dijkstra(int s,int e) //也可以和上面的模板一樣只寫帶一個引數的,在主函式中輸入e { priority_queue<HeapNode> Q; for(int i=0;i<n;i++) d[i]=INF; d[s]=0; Q.push(HeapNode(d[s],s)); memset(done,0,sizeof(done)); while(!Q.empty()) { HeapNode x= Q.top(); Q.pop(); int u =x.u; if(done[u]) continue; done[u]=true; for(int i=0;i<G[u].size();i++) { Edge &e=edges[G[u][i]]; if(d[e.to] > d[u]+e.dist) { d[e.to] = d[u]+e.dist; Q.push(HeapNode(d[e.to],e.to)); } } } if(d[e] == INF) //這裡需要特判一下 return -1; return d[e]; } }DJ; int main() { while(scanf("%d%d",&n,&m)==2) { DJ.init(n); for(int i=0;i<m;i++) { int u,v,d; scanf("%d%d%d",&u,&v,&d); DJ.AddEdge(u,v,d); DJ.AddEdge(v,u,d); } int s,e; scanf("%d%d",&s,&e); printf("%d\n",DJ.dijkstra(s,e) ); } return 0; }