总结树的遍历(致傻b的我)

树的前序(preorder)遍历:
iterative:
用stack,依次放右节点,左节点
python代码:

class Solution:
    """
    @param root: A Tree
    @return: Preorder in ArrayList which contains node values.
    """
    def preorderTraversal(self, root):
        # write your code here
        res = []
        stack = []
        if not root:
            return res
        stack.append(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

recursive:
就很简单啦,res添加当前节点,dfs访问左节点,dfs访问右节点。

class Solution:
    """
    @param root: A Tree
    @return: Preorder in ArrayList which contains node values.
    """
    def preorderTraversal(self, root):
        # write your code here
        res = []
        self.dfs(root, res)
        return res
    def dfs(self, node, res):
        if node:
            res.append(node.val)
            self.dfs(node.left, res)
            self.dfs(node.right, res)
        return

树的中序遍历(inorder): 左根右
iterative:
同样是用stack,从根开始依次把左边节点往栈里推,直到节点为空,栈pop,放入result list中,再把当前节点设置为pop出的节点的右节点,再执行上述操作。最后直到栈为空

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
"""

class Solution:
    """
    @param root: A Tree
    @return: Inorder in ArrayList which contains node values.
    """
    def inorderTraversal(self, root):
        # write your code here
        stack = []
        res = []
        if not root:
            return []
        
        cur = root
        while cur or stack:
            while cur:
                stack.append(cur)
                cur = cur.left
            node = stack.pop()
            res.append(node.val)
            cur = node.right
        return res

recursive:

class Solution:
    """
    @param root: A Tree
    @return: Inorder in ArrayList which contains node values.
    """
    def inorderTraversal(self, root):
        # write your code here
        res = []
        self.dfs(root, res)
        return res
        
    def dfs(self, node, res):
        if node:
            self.dfs(node.left, res)
            res.append(node.val)
            self.dfs(node.right, res)

树的后序遍历(post order):
iterative:

class Solution:
    """
    @param root: A Tree
    @return: Postorder in ArrayList which contains node values.
    """
    def postorderTraversal(self, root):
        # write your code here
        res = []
        stack = []
        if not root:
            return res
        stack.append(root)
        while stack:
            node = stack.pop()
            res.append(node.val)
            if node.left:
                stack.append(node.left)
            if node.right:
                stack.append(node.right)
        res.reverse()
        return res
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 上一篇文章讲述了树的概念, 特征以及分类, 旨在让我们理解什么是树, 树的一些常用的概念是什么,树的分类有哪些等。...
    DevCW阅读 2,084评论 4 10
  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,787评论 0 33
  • 编译环境:python v3.5.0, mac osx 10.11.4 前述内容: 线性表 队列 堆栈 线性结构...
    掷骰子的求阅读 2,520评论 1 7
  • 曲从雨中吟, 伤由月下生。 少思才少苦, 多愁总多心。 --2017...
    烽火煤阅读 169评论 0 1
  • 好吧,我事后诸葛亮一下,如果是定投的话,真的不必在意当前价格。尤其是购买的时候不要定价购买,而应当是现价购买。
    对酒几何阅读 179评论 0 0