| 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 |  | 
|---|