Trie: Add exist member function

akshitgrover · May 2, 2019 · 4e02b21 · 4e02b21
1 parent 9c2be47
commit 4e02b21
Show file tree

Hide file tree

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

diff --git a/trie.hpp b/trie.hpp
@@ -26,3 +26,23 @@ void trie<T>::insert(std::string key, T val) {
   node->update(val);
   node->markEnd();
 }
+
+template <typename T>
+bool trie<T>::exist(std::string key) {
+  int ascii;
+  bool res = true;
+  tnode<T>* node = this->root;
+  sIter(key, it) {
+    ascii = (int)*it;
+    if (node->getChild(ascii) == nullptr) {
+      res = false;
+      break;
+    } else {
+      node = node->getChild(ascii);
+    }
+  }
+  if (!node->isEnd()) {
+    res = false;
+  }
+  return res;
+}