【Python】【LeetCode.144】二叉树的前序遍历

最新推荐文章于 2024-10-03 21:46:55 发布

养猪业余户

最新推荐文章于 2024-10-03 21:46:55 发布

阅读量140

点赞数

文章标签：二叉树前序遍历递归迭代算法

本文链接：https://blog.csdn.net/weixin_40207523/article/details/119858908

版权

给你二叉树的根节点 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

首先想到的当然是喜闻乐见的递归

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
    	if not root:
    		return []
    	return [root.val] + self.preorderTraversal(val.left) + self.preorderTraversal(val.right)

由于递归过于简单，往往要求用迭代方式求解

# class TreeNode:
	# def __init__(self, val=0, left=None, right=None):
		# self.val = val
		# self.left = left
		# self.right = right
class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
    	res = []
    	stack = []
    	cur = root
    	while stack or cur:
    		while cur:
    			res.append(cur.val)
    			stack.append(cur)
    			cur = cur.left
    		cur = stack.pop()
    		cur = cur.right
    	return res