二叉树的中序遍历(python):递归+迭代

94 LeetCode 二叉树的中序遍历

"""
给定一个二叉树,返回它的中序遍历。

示例:
输入: [1,null,2,3]
   1
    \
     2
    /
   3

输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?

"""


定义树结构:

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

列表创建二叉树:

def Create_tree(root,llist,i):
    #创建过程也是从根开始a,创左子树b,再创b的左子树,如果b的左子树为空,返回none。
    #再接着创建b的右子树,
    if i

递归+迭代:

from typing import List
class Solution:
    def inorderTraversal(self, root):
        res = []
        def helper(root):
            if not root: return None
            helper(root.left)
            res.append(root.val)
            helper(root.right)
        helper(root)
        return res
    

迭代:

# 中序打印二叉树(非递归)
    def inOrderTraverse2(self,root):
        res = []
        stack = []
        # 用p当做指针
        p = root
        while p or stack:
            # 把左子树压入栈中
            while p:
                stack.append(p)
                p = p.left
            # 输出 栈顶元素
            p = stack.pop()
            res.append(p.val)
            # 看右子树
            p = p.right
        return res


测试代码:

if __name__=="__main__":
    s = Solution()
    root = ['1','2','3',None,'4','5']
    root = Create_tree(None,root,0)
    r = s.inorderTraversal(root)
    print(r)

 

你可能感兴趣的:(算法设计,python学习,心得体会)