Skip to content

Commit 62e604d

Browse files
committed
yield from tree inorder
1 parent 47b46b1 commit 62e604d

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

docs/14_树与二叉树/btree.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,12 @@ def inorder_trav(self, subtree):
8585
print(subtree.data)
8686
self.inorder_trav(subtree.right)
8787

88+
def yield_inorder(self, subtree): # for val in yield_inorder(root): print(val)
89+
if subtree:
90+
yield from self.inorder(subtree.left)
91+
yield subtree.val
92+
yield from self.inorder(subtree.right)
93+
8894
def reverse(self, subtree):
8995
if subtree is not None:
9096
subtree.left, subtree.right = subtree.right, subtree.left

0 commit comments

Comments
 (0)