本篇重要参考资料:
1.(点此跳转)。这篇解析,不仅简单介绍了二叉树这种数据结构,还详细罗列了各种不同遍历的方法,对于我使用的教材是个很好的补充。
2. (点此跳转) 这篇解析配上了动画,讲清楚了莫里斯遍历的方法。
题目
144. 二叉树的前序遍历
难度:中等
题目分析: 对于树的遍历问题,最简单最快的方法便是递归。另外,结合栈的使用,我们也可以通过迭代的方式树的遍历。
树有三种遍历类型,其中前序遍历是最为简单的。
1. 解法一:递归解法
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
res = []
def dfs(root):
if not root:
return
res.append(root.val)
dfs(root.left)
dfs(root.right)
dfs(root)
return res
1.1 运行结果:
1.2 分析:
存储结果的列表 res 也可以以参数的方式传入。
解法二:迭代的方法
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
# 喜欢答案一种简洁的写法,36ms,果然代码要能统一处理,会提高速度
res, stack = [], [root]
while stack:
node = stack.pop()
if node: # 非空才进行操作
res.append(node.val)
stack.append(node.right)
stack.append(node.left)
return res
2.1 运行结果:
2.2 分析:
上述的写法很简洁,不过最后每个叶子节点的左右空子树都会被存储进栈,浪费空间。
另一种方法是进栈前进行判断,不过其实只是语句多了一个if,但是时间上,应该不会有更多的判断。
3. 解法三:迭代,空节点不进栈
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def preorderTraversal(self, 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
3.1 运行结果:
3.2 分析:
这里只需要增加语句检查根节点是否为空,就可以保证判断的次数不增加,同时节省栈的空间。
速度依然很快。
4. 题外话
从5月中旬开始写这方面的题目,中间因为准备答辩,完全放下了写代码和整理……所以即使是简单的树的前序遍历问题,还是一时间不知道怎么下手,代码的练习,真的不能荒废。
另外,为了加快完成一篇博客的时间,接下来的博客主要呈现代码,只有特别重要的地方进行记录和总结。