代码随想录 -- 二叉树 -- 二叉树的迭代遍历

前序

遍历顺序为中左右，定义一个栈 stack，一个数组 res 存放最终结果。

注意：由于栈是后进先出，所以要按照右左来进栈。

144. 二叉树的前序遍历 - 力扣（LeetCode）

class Solution(object):
    def preorderTraversal(self, root):
        stack=[]
        res=[]
        node=root
        stack.append(node)
        while len(stack)!=0:
            cur=stack.pop()
            if cur!=None:
                res.append(cur.val)
                stack.append(cur.right)
                stack.append(cur.left)
        return res

后序

与前序类似，只需将中左右中的左右调换顺序，即中右左，再将 res 反转就能得到左右中的遍历顺序。

145. 二叉树的后序遍历 - 力扣（LeetCode）

class Solution(object):
    def postorderTraversal(self, root):
        stack=[]
        res=[]
        node=root
        stack.append(node)
        while len(stack)!=0:
            cur=stack.pop()
            if cur!=None:
                res.append(cur.val)
                stack.append(cur.left)
                stack.append(cur.right)
        res.reverse()
        return res

中序

在使用栈、数组的基础上再借助指针。

94. 二叉树的中序遍历 - 力扣（LeetCode）

class Solution(object):
    def inorderTraversal(self, root):
        if root==None:
            return
        node=root
        stack=[]
        res=[]
        while len(stack)!=0 or node!=None:
            if node!=None:
                stack.append(node)
                node=node.left
            else:
                node=stack.pop()
                res.append(node.val)
                node=node.right
        return res