树的遍历(前中后层)python

中序遍历:

#左中右
#递归实现
def inorderTraversal(self, root):
    if root==None:
        return []
    return self.inorderTraversal(root.left)+[root.val]+self.inorderTraversal(self.right)

#遍历实现
def inorderTraversal(self, root):

        stack=[]
        sol=[]
        curr=root
        while curr or stack:
                  #if curr 不为None,则进行栈添加,然后获取其左节点(存储的都为左和中节点)
                if curr:
                    stack.append(curr)
                    curr=curr.left        
                else:
    #左节点为空,才开始进行下一步,则进行添加上一个节点的取值

                    curr=stack.pop()
                    sol.append(curr.val)
                    curr=curr.right
        return sol                    

def preorderTraversal(self, root):
    if root==None:
        return []
    stack=[]
    sol=[]
    curr=root
    while curr or stack:
        if curr:
            sol.append(curr.val)
            stack.append(curr.right)
            curr=curr.left
#如果当前节点的左节点为None,则返回上一个节点的右节点
        else:
            curr=stack.pop()
    return sol