python实现前序遍历
时间: 2023-12-04 19:37:40 浏览: 75
前序遍历是二叉树遍历的一种方式,它的遍历顺序是先遍历根节点,然后遍历左子树,最后遍历右子树。下面是Python实现前序遍历的代码:
```
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def preorderTraversal(root: TreeNode) -> List[int]:
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
res.append(node.val)
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
return res
```
这段代码中,我们定义了一个TreeNode类来表示二叉树的节点,其中包含节点的值、左子树和右子树。preorderTraversal函数接收一个根节点作为参数,返回前序遍历的结果。我们使用一个栈来辅助遍历,首先将根节点入栈,然后每次从栈中弹出一个节点,将其值加入结果列表中,并将其右子树和左子树依次入栈。最后返回结果列表即可。
阅读全文