update 树的层次遍历

yinwoods · yinwoods · commit 947c33d73bbb · 2018-03-14T22:56:44.000+08:00
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:
+        row = [kid for item in row for kid in (item.left, item.right) if kid]
+        print(row)
 
 ```
 
