1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
mod impl_base_graph;
pub use self::impl_base_graph::*;
use core::*;
#[derive(Clone, Debug)]
pub struct AdjListGraph<V,W> {
edges: Vec<Vec<(usize,W)>>,
values:Vec<V>,
}
impl<V,W> AdjListGraph<V,W>
where
V: Copy + Eq,
W: Copy + Eq
{
pub fn new(values: Vec<V>, edges: Vec<(usize, usize,W)>) -> Option<AdjListGraph<V,W>> {
let mut g = AdjListGraph{ edges: Vec::new(), values: values };
for &(source, sink, _) in &edges {
if source >= g.values.len() || sink >= g.values.len(){
return None;
}
}
for _ in 0..g.values.len(){
g.edges.push(Vec::new());
}
for &(source, sink,weight) in &edges {
g.edges[source].push((sink,weight));
}
Some(g)
}
fn get_index(&self, v: V) -> Option<usize>{
self.values.iter().position(|ref value| **value == v)
}
fn get_value(&self, i: usize) -> Option<V>{
if i < self.values.len() {
Some(self.values[i].clone())
}else {
None
}
}
fn if_valid_edge<F>(&mut self, e:BaseEdge<V,W>, cont: F) -> Result<(), ()>
where F: Fn(&mut Self,usize, usize, W)-> Result<(),()>
{
if let (Some(source_i), Some(sink_i))
= (self.get_index(e.source), self.get_index(e.sink))
{
return cont(self, source_i, sink_i, e.weight);
}
Err(())
}
}