问题描述
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,2,3]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [1] 输出:[1]
示例 4:
输入:root = [1,2]
输出:[1,2]
示例 5:
输入:root = [1,null,2] 输出:[1,2]
提示:
树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析及代码实现
前序遍历:
class Solution:
def preorderTraversal(self, root: TreeNode) -> List[int]:
def dfs(root):
if not root:
return
res.append(root.val)
dfs(root.left)
dfs(root.right)
res = []
dfs(root)
return res
迭代:
利用栈来进行class Solution:
先入栈右孩子,再入栈左孩子,这样在出栈时就先出左孩子,也是按照根节点–左孩子–右孩子的顺序
def preorderTraversal(self, root: TreeNode) -> List[int]:
if not root:
return []
stack = [root]
res = []
while stack:
node = stack.pop()
if node:
res.append(node.val)
if node.right: # 先将右孩子入栈
stack.append(node.right)
if node.left: # 再将左孩子入栈
stack.append(node.left)
return res
遍历方式
如果按照 根节点 -> 左孩子 -> 右孩子 的方式遍历,即「前序遍历」,每次先遍历根节点,遍历结果为 1 2 4 5 3 6 7;
同理,如果按照 左孩子 -> 根节点 -> 右孩子 的方式遍历,即「中序序遍历」,遍历结果为 4 2 5 1 6 3 7;
如果按照 左孩子 -> 右孩子 -> 根节点 的方式遍历,即「后序序遍历」,遍历结果为 4 5 2 6 7 3 1;
最后,层次遍历就是按照每一层从左向右的方式进行遍历,遍历结果为 1 2 3 4 5 6 7。