... | ... |
@@ -111,223 +111,222 @@ |
111 | 111 |
template <typename In, typename Enable = void> |
112 | 112 |
struct SequenceInputIndicator { |
113 | 113 |
static const bool value = false; |
114 | 114 |
}; |
115 | 115 |
|
116 | 116 |
template <typename In> |
117 | 117 |
struct SequenceInputIndicator<In, |
118 | 118 |
typename exists<typename In::value_type::first_type>::type> { |
119 | 119 |
static const bool value = true; |
120 | 120 |
}; |
121 | 121 |
|
122 | 122 |
template <typename In, typename Enable = void> |
123 | 123 |
struct MapInputIndicator { |
124 | 124 |
static const bool value = false; |
125 | 125 |
}; |
126 | 126 |
|
127 | 127 |
template <typename In> |
128 | 128 |
struct MapInputIndicator<In, |
129 | 129 |
typename exists<typename In::Value>::type> { |
130 | 130 |
static const bool value = true; |
131 | 131 |
}; |
132 | 132 |
|
133 | 133 |
template <typename In, typename Enable = void> |
134 | 134 |
struct SequenceOutputIndicator { |
135 | 135 |
static const bool value = false; |
136 | 136 |
}; |
137 | 137 |
|
138 | 138 |
template <typename Out> |
139 | 139 |
struct SequenceOutputIndicator<Out, |
140 | 140 |
typename exists<typename Out::value_type>::type> { |
141 | 141 |
static const bool value = true; |
142 | 142 |
}; |
143 | 143 |
|
144 | 144 |
template <typename Out, typename Enable = void> |
145 | 145 |
struct MapOutputIndicator { |
146 | 146 |
static const bool value = false; |
147 | 147 |
}; |
148 | 148 |
|
149 | 149 |
template <typename Out> |
150 | 150 |
struct MapOutputIndicator<Out, |
151 | 151 |
typename exists<typename Out::Value>::type> { |
152 | 152 |
static const bool value = true; |
153 | 153 |
}; |
154 | 154 |
|
155 | 155 |
template <typename In, typename InEnable = void> |
156 | 156 |
struct KruskalValueSelector {}; |
157 | 157 |
|
158 | 158 |
template <typename In> |
159 | 159 |
struct KruskalValueSelector<In, |
160 | 160 |
typename enable_if<SequenceInputIndicator<In>, void>::type> |
161 | 161 |
{ |
162 | 162 |
typedef typename In::value_type::second_type Value; |
163 | 163 |
}; |
164 | 164 |
|
165 | 165 |
template <typename In> |
166 | 166 |
struct KruskalValueSelector<In, |
167 | 167 |
typename enable_if<MapInputIndicator<In>, void>::type> |
168 | 168 |
{ |
169 | 169 |
typedef typename In::Value Value; |
170 | 170 |
}; |
171 | 171 |
|
172 | 172 |
template <typename Graph, typename In, typename Out, |
173 | 173 |
typename InEnable = void> |
174 | 174 |
struct KruskalInputSelector {}; |
175 | 175 |
|
176 | 176 |
template <typename Graph, typename In, typename Out, |
177 | 177 |
typename InEnable = void> |
178 | 178 |
struct KruskalOutputSelector {}; |
179 | 179 |
|
180 | 180 |
template <typename Graph, typename In, typename Out> |
181 | 181 |
struct KruskalInputSelector<Graph, In, Out, |
182 | 182 |
typename enable_if<SequenceInputIndicator<In>, void>::type > |
183 | 183 |
{ |
184 | 184 |
typedef typename In::value_type::second_type Value; |
185 | 185 |
|
186 | 186 |
static Value kruskal(const Graph& graph, const In& in, Out& out) { |
187 | 187 |
return KruskalOutputSelector<Graph, In, Out>:: |
188 | 188 |
kruskal(graph, in, out); |
189 | 189 |
} |
190 | 190 |
|
191 | 191 |
}; |
192 | 192 |
|
193 | 193 |
template <typename Graph, typename In, typename Out> |
194 | 194 |
struct KruskalInputSelector<Graph, In, Out, |
195 | 195 |
typename enable_if<MapInputIndicator<In>, void>::type > |
196 | 196 |
{ |
197 | 197 |
typedef typename In::Value Value; |
198 | 198 |
static Value kruskal(const Graph& graph, const In& in, Out& out) { |
199 | 199 |
typedef typename In::Key MapArc; |
200 | 200 |
typedef typename In::Value Value; |
201 | 201 |
typedef typename ItemSetTraits<Graph, MapArc>::ItemIt MapArcIt; |
202 | 202 |
typedef std::vector<std::pair<MapArc, Value> > Sequence; |
203 | 203 |
Sequence seq; |
204 | 204 |
|
205 | 205 |
for (MapArcIt it(graph); it != INVALID; ++it) { |
206 | 206 |
seq.push_back(std::make_pair(it, in[it])); |
207 | 207 |
} |
208 | 208 |
|
209 | 209 |
std::sort(seq.begin(), seq.end(), PairComp<Sequence>()); |
210 | 210 |
return KruskalOutputSelector<Graph, Sequence, Out>:: |
211 | 211 |
kruskal(graph, seq, out); |
212 | 212 |
} |
213 | 213 |
}; |
214 | 214 |
|
215 | 215 |
template <typename T> |
216 | 216 |
struct RemoveConst { |
217 | 217 |
typedef T type; |
218 | 218 |
}; |
219 | 219 |
|
220 | 220 |
template <typename T> |
221 | 221 |
struct RemoveConst<const T> { |
222 | 222 |
typedef T type; |
223 | 223 |
}; |
224 | 224 |
|
225 | 225 |
template <typename Graph, typename In, typename Out> |
226 | 226 |
struct KruskalOutputSelector<Graph, In, Out, |
227 | 227 |
typename enable_if<SequenceOutputIndicator<Out>, void>::type > |
228 | 228 |
{ |
229 | 229 |
typedef typename In::value_type::second_type Value; |
230 | 230 |
|
231 | 231 |
static Value kruskal(const Graph& graph, const In& in, Out& out) { |
232 | 232 |
typedef LoggerBoolMap<typename RemoveConst<Out>::type> Map; |
233 | 233 |
Map map(out); |
234 | 234 |
return _kruskal_bits::kruskal(graph, in, map); |
235 | 235 |
} |
236 | 236 |
|
237 | 237 |
}; |
238 | 238 |
|
239 | 239 |
template <typename Graph, typename In, typename Out> |
240 | 240 |
struct KruskalOutputSelector<Graph, In, Out, |
241 | 241 |
typename enable_if<MapOutputIndicator<Out>, void>::type > |
242 | 242 |
{ |
243 | 243 |
typedef typename In::value_type::second_type Value; |
244 | 244 |
|
245 | 245 |
static Value kruskal(const Graph& graph, const In& in, Out& out) { |
246 | 246 |
return _kruskal_bits::kruskal(graph, in, out); |
247 | 247 |
} |
248 | 248 |
}; |
249 | 249 |
|
250 | 250 |
} |
251 | 251 |
|
252 | 252 |
/// \ingroup spantree |
253 | 253 |
/// |
254 | 254 |
/// \brief Kruskal algorithm to find a minimum cost spanning tree of |
255 | 255 |
/// a graph. |
256 | 256 |
/// |
257 | 257 |
/// This function runs Kruskal's algorithm to find a minimum cost |
258 | 258 |
/// spanning tree. |
259 | 259 |
/// Due to some C++ hacking, it accepts various input and output types. |
260 | 260 |
/// |
261 | 261 |
/// \param g The graph the algorithm runs on. |
262 | 262 |
/// It can be either \ref concepts::Digraph "directed" or |
263 | 263 |
/// \ref concepts::Graph "undirected". |
264 | 264 |
/// If the graph is directed, the algorithm consider it to be |
265 | 265 |
/// undirected by disregarding the direction of the arcs. |
266 | 266 |
/// |
267 | 267 |
/// \param in This object is used to describe the arc/edge costs. |
268 | 268 |
/// It can be one of the following choices. |
269 | 269 |
/// - An STL compatible 'Forward Container' with |
270 | 270 |
/// <tt>std::pair<GR::Arc,X></tt> or |
271 | 271 |
/// <tt>std::pair<GR::Edge,X></tt> as its <tt>value_type</tt>, where |
272 | 272 |
/// \c X is the type of the costs. The pairs indicates the arcs/edges |
273 | 273 |
/// along with the assigned cost. <em>They must be in a |
274 | 274 |
/// cost-ascending order.</em> |
275 | 275 |
/// - Any readable arc/edge map. The values of the map indicate the |
276 | 276 |
/// arc/edge costs. |
277 | 277 |
/// |
278 | 278 |
/// \retval out Here we also have a choice. |
279 | 279 |
/// - It can be a writable \c bool arc/edge map. After running the |
280 | 280 |
/// algorithm it will contain the found minimum cost spanning |
281 | 281 |
/// tree: the value of an arc/edge will be set to \c true if it belongs |
282 | 282 |
/// to the tree, otherwise it will be set to \c false. The value of |
283 | 283 |
/// each arc/edge will be set exactly once. |
284 | 284 |
/// - It can also be an iteraror of an STL Container with |
285 | 285 |
/// <tt>GR::Arc</tt> or <tt>GR::Edge</tt> as its |
286 | 286 |
/// <tt>value_type</tt>. The algorithm copies the elements of the |
287 | 287 |
/// found tree into this sequence. For example, if we know that the |
288 | 288 |
/// spanning tree of the graph \c g has say 53 arcs, then we can |
289 | 289 |
/// put its arcs into an STL vector \c tree with a code like this. |
290 | 290 |
///\code |
291 | 291 |
/// std::vector<Arc> tree(53); |
292 | 292 |
/// kruskal(g,cost,tree.begin()); |
293 | 293 |
///\endcode |
294 | 294 |
/// Or if we don't know in advance the size of the tree, we can |
295 | 295 |
/// write this. |
296 | 296 |
///\code |
297 | 297 |
/// std::vector<Arc> tree; |
298 | 298 |
/// kruskal(g,cost,std::back_inserter(tree)); |
299 | 299 |
///\endcode |
300 | 300 |
/// |
301 | 301 |
/// \return The total cost of the found spanning tree. |
302 | 302 |
/// |
303 |
/// \warning If Kruskal runs on an be consistent of using the same |
|
304 |
/// Arc type for input and output. |
|
305 |
/// |
|
303 |
/// \note If the input graph is not (weakly) connected, a spanning |
|
304 |
/// forest is calculated instead of a spanning tree. |
|
306 | 305 |
|
307 | 306 |
#ifdef DOXYGEN |
308 | 307 |
template <class Graph, class In, class Out> |
309 | 308 |
Value kruskal(GR const& g, const In& in, Out& out) |
310 | 309 |
#else |
311 | 310 |
template <class Graph, class In, class Out> |
312 | 311 |
inline typename _kruskal_bits::KruskalValueSelector<In>::Value |
313 | 312 |
kruskal(const Graph& graph, const In& in, Out& out) |
314 | 313 |
#endif |
315 | 314 |
{ |
316 | 315 |
return _kruskal_bits::KruskalInputSelector<Graph, In, Out>:: |
317 | 316 |
kruskal(graph, in, out); |
318 | 317 |
} |
319 | 318 |
|
320 | 319 |
|
321 | 320 |
|
322 | 321 |
|
323 | 322 |
template <class Graph, class In, class Out> |
324 | 323 |
inline typename _kruskal_bits::KruskalValueSelector<In>::Value |
325 | 324 |
kruskal(const Graph& graph, const In& in, const Out& out) |
326 | 325 |
{ |
327 | 326 |
return _kruskal_bits::KruskalInputSelector<Graph, In, const Out>:: |
328 | 327 |
kruskal(graph, in, out); |
329 | 328 |
} |
330 | 329 |
|
331 | 330 |
} //namespace lemon |
332 | 331 |
|
333 | 332 |
#endif //LEMON_KRUSKAL_H |
0 comments (0 inline)