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 };
		
		//Validate all edges point to vertices
		for &(source, sink, _) in &edges {
			if source >= g.values.len() || sink >= g.values.len(){
				return None;
			}
		}
		
		//Initialise adjacency list
		for _ in 0..g.values.len(){
			g.edges.push(Vec::new());
		}
		
		//Insert each edge
		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(())
	}
}