题目链接:
力扣_144.二叉树的前序遍历
二叉树的前序遍历的方式是根左右的顺序,那么优先选取递归的方式来做,之后再考虑是否可以用迭代的方式来完成
1.递归
一般很顺的递归的前序遍历都是print的方式,不需要考虑保存结点值,就能很快写出很基础的代码:
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
if not root:
return
print(root.val)
self.preorderTraversal(root.left)
self.preorderTraversal(root.right)
这个是一个很原始的模板,现在题目要求要返回这个树的前序遍历的结点值的数组,那么我们需要用一个数组来保存这个结果,就需要在这个模板上进行对应的修改
怎么修改呢?我们要用一个list
来不停的append
新的结点值,那么这个方法的入参就需要传入这样的一个list
,并且这个方法返回的值也应该是这个list
,那么我们就应该新写一方法来承载这两个入参,原函数就应该调用这个新的方法,传入root和一个初始化的数组res,然后在新的方法里更新这个res,在原方法里最终返回这个数组,(这个数组传入新函数之后一直在更新,所以不需要用一个返回值来接收,可以直接返回这个变量),
然后应该在新的方法里面去递归遍历并传入更新后的res,把print的那一行改为累加结点值res.append(root.val)
以达到保存所有结点值的效果
那么我们现在基本框架就搭建好了:
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
# if not root 这一步在preOrder里面做了
res = []
self.preOrder(root, res)
return res
def preOrder(self, root, res):
'''对root的判空其实是在这里做了,所以原方法就不需要对root再一次判空'''
if not root:
return res
res.