Update 208._implement_trie_(prefix_tree).md

208._implement_trie_(prefix_tree).md

@@ -56,10 +56,11 @@ class Trie(object):
         :rtype: bool
         """
         node = self.root
-        for letter in word:
-            node = node.childs.get(letter)
-            if node is None:
+        for i in word:
+            child = node.childs.get(i)
+            if child is None:
                 return False
+            node = child
         return node.isWord
         
 
@@ -72,9 +73,10 @@ class Trie(object):
         """
         node = self.root
         for letter in prefix:
-            node = node.childs.get(letter)
-            if node is None:
+            child = node.childs.get(letter)
+            if child is None:
                 return False
+            node = child
         return True