这里推荐这三道题先熟悉二叉树的三种遍历方式
144. 二叉树的前序遍历(中->左->右)
根左右。前序遍历首先访问根结点然后遍历左子树,最后遍历右子树。在遍历左、右子树时,仍然先访问根节点,然后遍历左子树,最后遍历右子树。
若二叉树为空则结束返回,否则:
(1)访问根结点。
(2)前序遍历左子树。
(3)前序遍历右子树 。
来自前序、中序、后续遍历二叉树 - 知乎 (zhihu.com)
def dfs(root):
if not root:
return []
left = dfs(root.left)
right = dfs(root.right)
return [root.val] + left + right
94. 二叉树的中序遍历
顺带注意,二叉树后序遍历结果为有序数组
def dfs(root):
if not root:
return []
left = dfs(root.left)
right = dfs(root.right)
return left + [root.val] + right
145. 二叉树的后序遍历
def dfs(root):
if not root:
return []
left = dfs(root.left)
right = dfs(root.right)
return left + right + [root.val]
写递归的时候严格按照递归三要素来写:
-
确定递归函数的参数和返回值: 确定哪些参数是递归的过程中需要处理的,那么就在递归函数里加上这个参数, 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。
-
确定终止条件: 写完了递归算法, 运行的时候,经常会遇到栈溢出的错误,就是没写终止条件或者终止条件写的不对,操作系统也是用一个栈的结构来保存每一层递归的信息,如果递归没有终止,操作系统的内存栈必然就会溢出。
-
确定单层递归的逻辑: 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。