Trie: Add begin memeber function

akshitgrover · May 5, 2019 · 6788428 · 6788428
1 parent bcc9ea0
commit 6788428
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/trie.h b/trie.h
@@ -11,6 +11,7 @@ class trie {
     void insert(std::string, T);
     bool exist(std::string);
     bool empty();
+    iterator begin();
   private:
     tnode<T>* root;
     int size;

diff --git a/trie.hpp b/trie.hpp
@@ -1,6 +1,7 @@
 #include <string>
 
 #include "_trie_util.h"
+#include "_trie_iterator.h"
 
 template <typename T>
 trie<T>::trie(): size(0) {
@@ -55,3 +56,9 @@ template <typename T>
 bool trie<T>::empty() {
   return this->size == 0;
 }
+
+template <typename T>
+typename trie<T>::iterator trie<T>::begin() {
+  trie_iterator<T> it = *(new trie_iterator<T>(this->root));
+  return ++it;
+}