1. 程式人生 > >二叉樹的後序遍歷(非遞歸方法)

二叉樹的後序遍歷(非遞歸方法)

lse lis res 遍歷 一個 int __init__ def pos

二叉樹的後序遍歷的話,利用stack進行非遞歸遍歷,要先訪問每個節點的左子樹,在訪問右子樹,然後訪問自身,

stack 一直循環,當棧頂節點(cur)的左右子樹都是None的時候,

        或者當pre節點不是None的時候,同時pre和cur節點的左右子節點中的一個相等的時候

        可以出棧並訪問

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

class Solution(object):
    def postorderTraversal(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if root ==None:
            return []
        stack=[root]
        pre=None
        result=[]
        while (len(stack)!=0):
            cur =stack[-1]
            if ((cur.left ==None and cur.right==None) or (pre!=None and (pre==cur.left or  pre== cur.right))):
                
                cur=stack.pop()
                result.append(cur.val)
                pre=cur
            else:
                if cur.right!=None:
                    stack.append(cur.right)
                if cur.left!=None:
                    stack.append(cur.left)
        return result

二叉樹的後序遍歷(非遞歸方法)