| 1 | #include <iostream> |
|---|
| 2 | |
|---|
| 3 | #include <lemon/list_graph.h> |
|---|
| 4 | #include <lemon/dijkstra.h> |
|---|
| 5 | |
|---|
| 6 | using namespace lemon; |
|---|
| 7 | |
|---|
| 8 | |
|---|
| 9 | int main (int, char*[]) |
|---|
| 10 | { |
|---|
| 11 | |
|---|
| 12 | typedef ListGraph Graph; |
|---|
| 13 | typedef Graph::Node Node; |
|---|
| 14 | typedef Graph::Edge Edge; |
|---|
| 15 | typedef Graph::EdgeMap<int> LengthMap; |
|---|
| 16 | |
|---|
| 17 | Graph g; |
|---|
| 18 | |
|---|
| 19 | //An example from Ahuja's book |
|---|
| 20 | |
|---|
| 21 | Node s=g.addNode(); |
|---|
| 22 | Node v2=g.addNode(); |
|---|
| 23 | Node v3=g.addNode(); |
|---|
| 24 | Node v4=g.addNode(); |
|---|
| 25 | Node v5=g.addNode(); |
|---|
| 26 | Node t=g.addNode(); |
|---|
| 27 | |
|---|
| 28 | Edge s_v2=g.addEdge(s, v2); |
|---|
| 29 | Edge s_v3=g.addEdge(s, v3); |
|---|
| 30 | Edge v2_v4=g.addEdge(v2, v4); |
|---|
| 31 | Edge v2_v5=g.addEdge(v2, v5); |
|---|
| 32 | Edge v3_v5=g.addEdge(v3, v5); |
|---|
| 33 | Edge v4_t=g.addEdge(v4, t); |
|---|
| 34 | Edge v5_t=g.addEdge(v5, t); |
|---|
| 35 | |
|---|
| 36 | LengthMap len(g); |
|---|
| 37 | |
|---|
| 38 | len.set(s_v2, 10); |
|---|
| 39 | len.set(s_v3, 10); |
|---|
| 40 | len.set(v2_v4, 5); |
|---|
| 41 | len.set(v2_v5, 8); |
|---|
| 42 | len.set(v3_v5, 5); |
|---|
| 43 | len.set(v4_t, 8); |
|---|
| 44 | len.set(v5_t, 8); |
|---|
| 45 | |
|---|
| 46 | std::cout << "The id of s is " << g.id(s)<< ", the id of t is " << g.id(t)<<"."<<std::endl; |
|---|
| 47 | |
|---|
| 48 | std::cout << "Dijkstra algorithm test..." << std::endl; |
|---|
| 49 | |
|---|
| 50 | Dijkstra<Graph, LengthMap> dijkstra_test(g,len); |
|---|
| 51 | |
|---|
| 52 | dijkstra_test.run(s); |
|---|
| 53 | |
|---|
| 54 | |
|---|
| 55 | std::cout << "The distance of node t from node s: " << dijkstra_test.dist(t)<<std::endl; |
|---|
| 56 | |
|---|
| 57 | std::cout << "The shortest path from s to t goes through the following nodes (the first one is t, the last one is s): "<<std::endl; |
|---|
| 58 | |
|---|
| 59 | for (Node v=t;v != s; v=dijkstra_test.predNode(v)){ |
|---|
| 60 | std::cout << g.id(v) << "<-"; |
|---|
| 61 | } |
|---|
| 62 | std::cout << g.id(s) << std::endl; |
|---|
| 63 | |
|---|
| 64 | |
|---|
| 65 | return 0; |
|---|
| 66 | } |
|---|
| 67 | |
|---|
| 68 | |
|---|
| 69 | |
|---|
| 70 | |
|---|
| 71 | |
|---|
| 72 | |
|---|
| 73 | |
|---|
| 74 | |
|---|
| 75 | |
|---|