| 1 | //======================================================================= |
|---|
| 2 | // Copyright 2001 Jeremy G. Siek, Andrew Lumsdaine, Lie-Quan Lee, |
|---|
| 3 | // |
|---|
| 4 | // Distributed under the Boost Software License, Version 1.0. (See |
|---|
| 5 | // accompanying file LICENSE_1_0.txt or copy at |
|---|
| 6 | // http://www.boost.org/LICENSE_1_0.txt) |
|---|
| 7 | //======================================================================= |
|---|
| 8 | #include <boost/config.hpp> |
|---|
| 9 | #include <iostream> |
|---|
| 10 | #include <boost/graph/adjacency_list.hpp> |
|---|
| 11 | #include <boost/graph/prim_minimum_spanning_tree.hpp> |
|---|
| 12 | |
|---|
| 13 | int |
|---|
| 14 | main() |
|---|
| 15 | { |
|---|
| 16 | using namespace boost; |
|---|
| 17 | typedef adjacency_list < vecS, vecS, undirectedS, |
|---|
| 18 | property<vertex_distance_t, int>, property < edge_weight_t, int > > Graph; |
|---|
| 19 | typedef std::pair < int, int >E; |
|---|
| 20 | const int num_nodes = 5; |
|---|
| 21 | E edges[] = { E(0, 2), E(1, 3), E(1, 4), E(2, 1), E(2, 3), |
|---|
| 22 | E(3, 4), E(4, 0) |
|---|
| 23 | }; |
|---|
| 24 | int weights[] = { 1, 1, 2, 7, 3, 1, 1 }; |
|---|
| 25 | #if defined(BOOST_MSVC) && BOOST_MSVC <= 1300 |
|---|
| 26 | Graph g(num_nodes); |
|---|
| 27 | property_map<Graph, edge_weight_t>::type weightmap = get(edge_weight, g); |
|---|
| 28 | for (std::size_t j = 0; j < sizeof(edges) / sizeof(E); ++j) { |
|---|
| 29 | graph_traits<Graph>::edge_descriptor e; bool inserted; |
|---|
| 30 | tie(e, inserted) = add_edge(edges[j].first, edges[j].second, g); |
|---|
| 31 | weightmap[e] = weights[j]; |
|---|
| 32 | } |
|---|
| 33 | #else |
|---|
| 34 | Graph g(edges, edges + sizeof(edges) / sizeof(E), weights, num_nodes); |
|---|
| 35 | property_map<Graph, edge_weight_t>::type weightmap = get(edge_weight, g); |
|---|
| 36 | #endif |
|---|
| 37 | std::vector < graph_traits < Graph >::vertex_descriptor > |
|---|
| 38 | p(num_vertices(g)); |
|---|
| 39 | |
|---|
| 40 | #if defined(BOOST_MSVC) && BOOST_MSVC <= 1300 |
|---|
| 41 | property_map<Graph, vertex_distance_t>::type distance = get(vertex_distance, g); |
|---|
| 42 | property_map<Graph, vertex_index_t>::type indexmap = get(vertex_index, g); |
|---|
| 43 | prim_minimum_spanning_tree |
|---|
| 44 | (g, *vertices(g).first, &p[0], distance, weightmap, indexmap, |
|---|
| 45 | default_dijkstra_visitor()); |
|---|
| 46 | #else |
|---|
| 47 | prim_minimum_spanning_tree(g, &p[0]); |
|---|
| 48 | #endif |
|---|
| 49 | |
|---|
| 50 | for (std::size_t i = 0; i != p.size(); ++i) |
|---|
| 51 | if (p[i] != i) |
|---|
| 52 | std::cout << "parent[" << i << "] = " << p[i] << std::endl; |
|---|
| 53 | else |
|---|
| 54 | std::cout << "parent[" << i << "] = no parent" << std::endl; |
|---|
| 55 | |
|---|
| 56 | return EXIT_SUCCESS; |
|---|
| 57 | } |
|---|