graph/example/kruskal.cpp
2000-12-09 19:22:30 +00:00

91 lines
3.0 KiB
C++

//=======================================================================
// Copyright 1997, 1998, 1999, 2000 University of Notre Dame.
// Authors: Andrew Lumsdaine, Lie-Quan Lee, Jeremy G. Siek
//
// This file is part of the Boost Graph Library
//
// You should have received a copy of the License Agreement for the
// Boost Graph Library along with the software; see the file LICENSE.
// If not, contact Office of Research, University of Notre Dame, Notre
// Dame, IN 46556.
//
// Permission to modify the code and to distribute modified code is
// granted, provided the text of this NOTICE is retained, a notice that
// the code was modified is included with the above COPYRIGHT NOTICE and
// with the COPYRIGHT NOTICE in the LICENSE file, and that the LICENSE
// file is distributed with the modified code.
//
// LICENSOR MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED.
// By way of example, but not limitation, Licensor MAKES NO
// REPRESENTATIONS OR WARRANTIES OF MERCHANTABILITY OR FITNESS FOR ANY
// PARTICULAR PURPOSE OR THAT THE USE OF THE LICENSED SOFTWARE COMPONENTS
// OR DOCUMENTATION WILL NOT INFRINGE ANY PATENTS, COPYRIGHTS, TRADEMARKS
// OR OTHER RIGHTS.
//=======================================================================
#include <boost/config.hpp>
#include <vector>
#include <iostream>
#include <iterator>
#include <boost/graph/adjacency_list.hpp>
#include <boost/graph/kruskal_min_spanning_tree.hpp>
/*
Sample output:
Print the edge in MST:
0 <--> 2 with weight of 1
3 <--> 4 with weight of 1
4 <--> 0 with weight of 1
1 <--> 3 with weight of 1
*/
int main(int , char* [])
{
using namespace boost;
using namespace std;
typedef property<edge_weight_t,int> weightp;
typedef adjacency_list< vecS, vecS, undirectedS,
no_property, weightp > Graph;
typedef Graph::edge_descriptor Edge;
typedef Graph::vertex_descriptor Vertex;
typedef std::pair<int,int> E;
const int num_nodes = 5;
E edges[] = { E(0,2),
E(1,1), E(1,3), E(1,4),
E(2,1), E(2,3),
E(3,4),
E(4,0), E(4,1) };
int weights[] = { 1,
2, 1, 2,
7, 3,
1,
1, 1};
Graph G(num_nodes, edges, edges + sizeof(edges)/sizeof(E), weights);
property_map<Graph, edge_weight_t>::type weight = get(edge_weight, G);
typedef std::vector<Edge> container;
std::vector<Edge> spanning_tree_edges;
spanning_tree_edges.reserve(num_vertices(G));
std::vector<Vertex> p(num_vertices(G));
std::vector<int> rank(num_vertices(G));
kruskal_minimum_spanning_tree(G, std::back_inserter(spanning_tree_edges),
&rank[0], &p[0]);
cout << "Print the edge in MST:" << endl;
for (std::vector<Edge>::iterator ei = spanning_tree_edges.begin();
ei != spanning_tree_edges.end(); ++ei) {
cout << source(*ei, G) << " <--> "
<< target(*ei, G)
<< " with weight of " << weight[*ei]
<< endl;
}
return 0;
}