123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347 |
- //
- //=======================================================================
- // Copyright 1997, 1998, 1999, 2000 University of Notre Dame.
- // Authors: Andrew Lumsdaine, Lie-Quan Lee, Jeremy G. Siek
- //
- // Distributed under the Boost Software License, Version 1.0. (See
- // accompanying file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt)
- //=======================================================================
- //
- #ifndef BOOST_GRAPH_STRONG_COMPONENTS_HPP
- #define BOOST_GRAPH_STRONG_COMPONENTS_HPP
- #include <stack>
- #include <boost/config.hpp>
- #include <boost/graph/depth_first_search.hpp>
- #include <boost/type_traits/conversion_traits.hpp>
- #include <boost/static_assert.hpp>
- #include <boost/graph/overloading.hpp>
- #include <boost/graph/detail/mpi_include.hpp>
- #include <boost/concept/assert.hpp>
- namespace boost
- {
- //==========================================================================
- // This is Tarjan's algorithm for strongly connected components
- // from his paper "Depth first search and linear graph algorithms".
- // It calculates the components in a single application of DFS.
- // We implement the algorithm as a dfs-visitor.
- namespace detail
- {
- template < typename ComponentMap, typename RootMap, typename DiscoverTime,
- typename Stack >
- class tarjan_scc_visitor : public dfs_visitor<>
- {
- typedef typename property_traits< ComponentMap >::value_type comp_type;
- typedef typename property_traits< DiscoverTime >::value_type time_type;
- public:
- tarjan_scc_visitor(ComponentMap comp_map, RootMap r, DiscoverTime d,
- comp_type& c_, Stack& s_)
- : c(c_)
- , comp(comp_map)
- , root(r)
- , discover_time(d)
- , dfs_time(time_type())
- , s(s_)
- {
- }
- template < typename Graph >
- void discover_vertex(
- typename graph_traits< Graph >::vertex_descriptor v, const Graph&)
- {
- put(root, v, v);
- put(comp, v, (std::numeric_limits< comp_type >::max)());
- put(discover_time, v, dfs_time++);
- s.push(v);
- }
- template < typename Graph >
- void finish_vertex(
- typename graph_traits< Graph >::vertex_descriptor v, const Graph& g)
- {
- typename graph_traits< Graph >::vertex_descriptor w;
- typename graph_traits< Graph >::out_edge_iterator ei, ei_end;
- for (boost::tie(ei, ei_end) = out_edges(v, g); ei != ei_end; ++ei)
- {
- w = target(*ei, g);
- if (get(comp, w) == (std::numeric_limits< comp_type >::max)())
- put(root, v,
- this->min_discover_time(get(root, v), get(root, w)));
- }
- if (get(root, v) == v)
- {
- do
- {
- w = s.top();
- s.pop();
- put(comp, w, c);
- put(root, w, v);
- } while (w != v);
- ++c;
- }
- }
- private:
- template < typename Vertex >
- Vertex min_discover_time(Vertex u, Vertex v)
- {
- return get(discover_time, u) < get(discover_time, v) ? u : v;
- }
- comp_type& c;
- ComponentMap comp;
- RootMap root;
- DiscoverTime discover_time;
- time_type dfs_time;
- Stack& s;
- };
- template < class Graph, class ComponentMap, class RootMap,
- class DiscoverTime, class P, class T, class R >
- typename property_traits< ComponentMap >::value_type strong_components_impl(
- const Graph& g, // Input
- ComponentMap comp, // Output
- // Internal record keeping
- RootMap root, DiscoverTime discover_time,
- const bgl_named_params< P, T, R >& params)
- {
- typedef typename graph_traits< Graph >::vertex_descriptor Vertex;
- BOOST_CONCEPT_ASSERT(
- (ReadWritePropertyMapConcept< ComponentMap, Vertex >));
- BOOST_CONCEPT_ASSERT((ReadWritePropertyMapConcept< RootMap, Vertex >));
- typedef typename property_traits< RootMap >::value_type RootV;
- BOOST_CONCEPT_ASSERT((ConvertibleConcept< RootV, Vertex >));
- BOOST_CONCEPT_ASSERT(
- (ReadWritePropertyMapConcept< DiscoverTime, Vertex >));
- typename property_traits< ComponentMap >::value_type total = 0;
- std::stack< Vertex > s;
- detail::tarjan_scc_visitor< ComponentMap, RootMap, DiscoverTime,
- std::stack< Vertex > >
- vis(comp, root, discover_time, total, s);
- depth_first_search(g, params.visitor(vis));
- return total;
- }
- //-------------------------------------------------------------------------
- // The dispatch functions handle the defaults for the rank and discover
- // time property maps.
- // dispatch with class specialization to avoid VC++ bug
- template < class DiscoverTimeMap > struct strong_comp_dispatch2
- {
- template < class Graph, class ComponentMap, class RootMap, class P,
- class T, class R >
- inline static typename property_traits< ComponentMap >::value_type
- apply(const Graph& g, ComponentMap comp, RootMap r_map,
- const bgl_named_params< P, T, R >& params, DiscoverTimeMap time_map)
- {
- return strong_components_impl(g, comp, r_map, time_map, params);
- }
- };
- template <> struct strong_comp_dispatch2< param_not_found >
- {
- template < class Graph, class ComponentMap, class RootMap, class P,
- class T, class R >
- inline static typename property_traits< ComponentMap >::value_type
- apply(const Graph& g, ComponentMap comp, RootMap r_map,
- const bgl_named_params< P, T, R >& params, param_not_found)
- {
- typedef
- typename graph_traits< Graph >::vertices_size_type size_type;
- size_type n = num_vertices(g) > 0 ? num_vertices(g) : 1;
- std::vector< size_type > time_vec(n);
- return strong_components_impl(g, comp, r_map,
- make_iterator_property_map(time_vec.begin(),
- choose_const_pmap(
- get_param(params, vertex_index), g, vertex_index),
- time_vec[0]),
- params);
- }
- };
- template < class Graph, class ComponentMap, class RootMap, class P, class T,
- class R, class DiscoverTimeMap >
- inline typename property_traits< ComponentMap >::value_type scc_helper2(
- const Graph& g, ComponentMap comp, RootMap r_map,
- const bgl_named_params< P, T, R >& params, DiscoverTimeMap time_map)
- {
- return strong_comp_dispatch2< DiscoverTimeMap >::apply(
- g, comp, r_map, params, time_map);
- }
- template < class RootMap > struct strong_comp_dispatch1
- {
- template < class Graph, class ComponentMap, class P, class T, class R >
- inline static typename property_traits< ComponentMap >::value_type
- apply(const Graph& g, ComponentMap comp,
- const bgl_named_params< P, T, R >& params, RootMap r_map)
- {
- return scc_helper2(g, comp, r_map, params,
- get_param(params, vertex_discover_time));
- }
- };
- template <> struct strong_comp_dispatch1< param_not_found >
- {
- template < class Graph, class ComponentMap, class P, class T, class R >
- inline static typename property_traits< ComponentMap >::value_type
- apply(const Graph& g, ComponentMap comp,
- const bgl_named_params< P, T, R >& params, param_not_found)
- {
- typedef typename graph_traits< Graph >::vertex_descriptor Vertex;
- typename std::vector< Vertex >::size_type n
- = num_vertices(g) > 0 ? num_vertices(g) : 1;
- std::vector< Vertex > root_vec(n);
- return scc_helper2(g, comp,
- make_iterator_property_map(root_vec.begin(),
- choose_const_pmap(
- get_param(params, vertex_index), g, vertex_index),
- root_vec[0]),
- params, get_param(params, vertex_discover_time));
- }
- };
- template < class Graph, class ComponentMap, class RootMap, class P, class T,
- class R >
- inline typename property_traits< ComponentMap >::value_type scc_helper1(
- const Graph& g, ComponentMap comp,
- const bgl_named_params< P, T, R >& params, RootMap r_map)
- {
- return detail::strong_comp_dispatch1< RootMap >::apply(
- g, comp, params, r_map);
- }
- } // namespace detail
- template < class Graph, class ComponentMap, class P, class T, class R >
- inline typename property_traits< ComponentMap >::value_type strong_components(
- const Graph& g, ComponentMap comp,
- const bgl_named_params< P, T, R >& params BOOST_GRAPH_ENABLE_IF_MODELS_PARM(
- Graph, vertex_list_graph_tag))
- {
- typedef typename graph_traits< Graph >::directed_category DirCat;
- BOOST_STATIC_ASSERT(
- (is_convertible< DirCat*, directed_tag* >::value == true));
- return detail::scc_helper1(
- g, comp, params, get_param(params, vertex_root_t()));
- }
- template < class Graph, class ComponentMap >
- inline typename property_traits< ComponentMap >::value_type strong_components(
- const Graph& g,
- ComponentMap comp BOOST_GRAPH_ENABLE_IF_MODELS_PARM(
- Graph, vertex_list_graph_tag))
- {
- typedef typename graph_traits< Graph >::directed_category DirCat;
- BOOST_STATIC_ASSERT(
- (is_convertible< DirCat*, directed_tag* >::value == true));
- bgl_named_params< int, int > params(0);
- return strong_components(g, comp, params);
- }
- template < typename Graph, typename ComponentMap, typename ComponentLists >
- void build_component_lists(const Graph& g,
- typename graph_traits< Graph >::vertices_size_type num_scc,
- ComponentMap component_number, ComponentLists& components)
- {
- components.resize(num_scc);
- typename graph_traits< Graph >::vertex_iterator vi, vi_end;
- for (boost::tie(vi, vi_end) = vertices(g); vi != vi_end; ++vi)
- components[component_number[*vi]].push_back(*vi);
- }
- } // namespace boost
- #include <queue>
- #include <vector>
- #include <boost/graph/transpose_graph.hpp>
- #include <boost/pending/indirect_cmp.hpp>
- #include <boost/graph/connected_components.hpp> // for components_recorder
- namespace boost
- {
- //==========================================================================
- // This is the version of strongly connected components from
- // "Intro. to Algorithms" by Cormen, Leiserson, Rivest, which was
- // adapted from "Data Structure and Algorithms" by Aho, Hopcroft,
- // and Ullman, who credit the algorithm to S.R. Kosaraju and M. Sharir.
- // The algorithm is based on computing DFS forests the graph
- // and its transpose.
- // This algorithm is slower than Tarjan's by a constant factor, uses
- // more memory, and puts more requirements on the graph type.
- template < class Graph, class DFSVisitor, class ComponentsMap,
- class DiscoverTime, class FinishTime, class ColorMap >
- typename property_traits< ComponentsMap >::value_type
- kosaraju_strong_components(
- Graph& G, ComponentsMap c, FinishTime finish_time, ColorMap color)
- {
- BOOST_CONCEPT_ASSERT((MutableGraphConcept< Graph >));
- // ...
- typedef typename graph_traits< Graph >::vertex_descriptor Vertex;
- typedef typename property_traits< ColorMap >::value_type ColorValue;
- typedef color_traits< ColorValue > Color;
- typename property_traits< FinishTime >::value_type time = 0;
- depth_first_search(G,
- make_dfs_visitor(stamp_times(finish_time, time, on_finish_vertex())),
- color);
- Graph G_T(num_vertices(G));
- transpose_graph(G, G_T);
- typedef typename property_traits< ComponentsMap >::value_type count_type;
- count_type c_count(0);
- detail::components_recorder< ComponentsMap > vis(c, c_count);
- // initialize G_T
- typename graph_traits< Graph >::vertex_iterator ui, ui_end;
- for (boost::tie(ui, ui_end) = vertices(G_T); ui != ui_end; ++ui)
- put(color, *ui, Color::white());
- typedef typename property_traits< FinishTime >::value_type D;
- typedef indirect_cmp< FinishTime, std::less< D > > Compare;
- Compare fl(finish_time);
- std::priority_queue< Vertex, std::vector< Vertex >, Compare > Q(fl);
- typename graph_traits< Graph >::vertex_iterator i, j, iend, jend;
- boost::tie(i, iend) = vertices(G_T);
- boost::tie(j, jend) = vertices(G);
- for (; i != iend; ++i, ++j)
- {
- put(finish_time, *i, get(finish_time, *j));
- Q.push(*i);
- }
- while (!Q.empty())
- {
- Vertex u = Q.top();
- Q.pop();
- if (get(color, u) == Color::white())
- {
- depth_first_visit(G_T, u, vis, color);
- ++c_count;
- }
- }
- return c_count;
- }
- } // namespace boost
- #include BOOST_GRAPH_MPI_INCLUDE(< boost / graph / distributed / strong_components.hpp >)
- #endif // BOOST_GRAPH_STRONG_COMPONENTS_HPP
|