From ba0cdb3e92c6b417eb5e47373ebfed2dff0340b0 Mon Sep 17 00:00:00 2001
From: wyy <wuyanyi09@gmail.com>
Date: Mon, 1 Jul 2013 23:42:03 +0800
Subject: [PATCH] add iterator and bak

---
 Trie.cpp | 17 +++++++++++++++--
 Trie.h   | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 64 insertions(+), 2 deletions(-)

diff --git a/Trie.cpp b/Trie.cpp
index f5eaa9d..de26d24 100644
--- a/Trie.cpp
+++ b/Trie.cpp
@@ -2,6 +2,16 @@
 
 namespace CppJieba
 {
+    Trie::iterator Trie::begin()
+    {
+        return Trie::iterator(_root);
+    }
+
+    Trie::iterator Trie::end()
+    {
+        return Trie::iterator(NULL);
+    }
+
     Trie::Trie()
     {
         _root = NULL;
@@ -167,8 +177,11 @@ using namespace CppJieba;
 int main()
 {
     Trie trie;
-    //trie.init("test/dict.txt");
-    trie.init("dict.txt");
+    trie.init("test/dict.txt");
+    cout<<trie.begin()->count<<endl;
+    
+    return 0;
+    //trie.init("dict.txt");
     //trie.display();
     //const char * utf = "B";
     //ChUnicode chUniStr[16];
diff --git a/Trie.h b/Trie.h
index 4656ee7..5a2b573 100644
--- a/Trie.h
+++ b/Trie.h
@@ -30,10 +30,59 @@ namespace CppJieba
         }
     };
 
+    struct TrieNodeIterator
+    {
+        TrieNode* ptNode;
+        
+        TrieNodeIterator():ptNode(NULL)
+        {
+        }
+
+        TrieNodeIterator(TrieNode* ptr):ptNode(NULL)
+        {
+            ptNode = ptr;
+        }
+
+        const int operator++(int)
+        {
+            return 1;
+        }
+
+        TrieNodeIterator& operator++()
+        {
+            return *this;
+        }
+
+        TrieNode& operator*() const
+        {
+            return *ptNode;
+        }
+
+        TrieNode* operator->() const
+        {
+            return ptNode;
+        }
+        
+        bool operator==(const TrieNodeIterator& x) const
+        {
+            return ptNode == x.ptNode;
+        }
+        
+        bool operator!=(const TrieNodeIterator& x) const
+        {
+            return ptNode != x.ptNode;
+        }
+    };
+
     class Trie
     {
         private:
             TrieNode* _root;
+        public:
+            typedef TrieNodeIterator iterator;
+        public:
+            iterator begin();
+            iterator end();
         public:
             Trie();
             ~Trie();