1 files changed, 35 insertions, 0 deletions

@@ -0,0 +1,35 @@
+#include "prefix_search.h"
+
+void prefix_search::add(std::string prefix)
+{
+  node* cur = ⊤
+  for (int c : prefix)
+  {
+    cur = &cur->children[c];
+  }
+  
+  cur->match = true;
+}
+
+int prefix_search::match(std::string in) const
+{
+  int ret = 0;
+  const node* cur = ⊤
+  for (int c : in)
+  {
+    if (cur->children.count(c) == 0)
+    {
+      return 0;
+    }
+    
+    cur = &cur->children.at(c);
+    ret++;
+    
+    if (cur->match)
+    {
+      return ret;
+    }
+  }
+  
+  return 0;
+}