tries 0.1.0

Rust implementations of Trie
Documentation 
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use hashbrown::HashMap;

#[derive(Debug, Clone)]
struct Node {
    children: HashMap<char, Node>,
    is_end: bool,
}

impl Node {
    fn new() -> Self {
        Self {
            children: HashMap::new(),
            is_end: false,
        }
    }
}

/// A simple Trie.
///
/// # Example
///
/// ```
/// use tries::Trie;
///
/// let mut trie = Trie::new();
/// trie.insert("happy");
/// trie.insert("happily");
///
/// assert!(trie.search("happy"));
/// assert!(trie.search("happily"));
/// assert!(!trie.search("hello"));
/// ```
#[derive(Debug, Clone)]
pub struct Trie {
    root: Node,
}

impl Trie {
    /// Create a Trie.
    pub fn new() -> Self {
        Self { root: Node::new() }
    }

    /// Insert a word in Trie.
    ///
    /// # Example
    ///
    /// ```
    /// use tries::Trie;
    ///
    /// let mut trie = Trie::new();
    ///
    /// trie.insert("hello");
    /// ```
    pub fn insert<S: AsRef<str>>(&mut self, word: S) {
        let mut current = &mut self.root;
        for c in word.as_ref().chars() {
            current = current.children.entry(c).or_insert(Node::new());
        }
        current.is_end = true;
    }

    /// Search a word in Trie.
    ///
    /// # Example
    ///
    /// ```
    /// use tries::Trie;
    ///
    /// let mut trie = Trie::new();
    ///
    /// trie.insert("hello");
    ///
    /// assert!(trie.search("hello"));
    /// assert!(!trie.search("ok"))
    /// ```
    pub fn search<S: AsRef<str>>(&mut self, word: S) -> bool {
        let mut current = &self.root;
        for c in word.as_ref().chars() {
            if let Some(node) = current.children.get(&c) {
                current = node;
            } else {
                return false;
            }
        }
        current.is_end
    }
}

impl Default for Trie {
    fn default() -> Self {
        Self::new()
    }
}