Merge pull request #48 from yinwoods/master

update 树的层次遍历
taizilongxu · Mar 22, 2018 · ef9685e · ef9685e
2 parents b1d68d9 + 76939ef
commit ef9685e
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/Readme.md b/Readme.md
@@ -1664,14 +1664,10 @@ tree = Node(1, Node(3, Node(7, Node(0)), Node(6)), Node(2, Node(5), Node(4)))
 ```python
 
 def lookup(root):
-    stack = [root]
-    while stack:
-        current = stack.pop(0)
-        print current.data
-        if current.left:
-            stack.append(current.left)
-        if current.right:
-            stack.append(current.right)
+    row = [root]
+    while row:
+        print(row)
+        row = [kid for item in row for kid in (item.left, item.right) if kid]
 
 ```