leetcode-145. 二叉树的后序遍历

145. 二叉树的后序遍历
     给定一个二叉树，返回它的 后序 遍历。

示例:

输入: [1,null,2,3]
1

2
/
3

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

# Definition for a binary tree node.
from typing import List


class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

n1 = TreeNode(1)
n2 = TreeNode(2)
n3 = TreeNode(3)
n1.right = n2
n2.left = n3
class Solution:
    def postorderTraversal(self, root: TreeNode) -> List[int]:

        # 非递归方法
        temp = []  # 保存结点
        ret = []  # 保存节点的值
        if root:
            temp.append(root)
        while temp:
            while temp[-1].left:  # 左子树存在，一直入栈
                temp.append(temp[-1].left)

            if temp[-1].right:
                temp.append(temp[-1].right)  # 左子树不存在，右子树存在入栈
            else:
                if not temp[-1].right:
                    # 右子树不存在  的话就是走到的叶节点
                    node = temp.pop(-1)
                    ret.append(node.val)
                    if temp:  # 断开节点
                        if temp[-1].left == node:
                            temp[-1].left = None
                        if temp[-1].right == node:
                            temp[-1].right = None

        return ret

s = Solution()
print(s.postorderTraversal(n1))



        # 递归方法
        # if root:
        #     ret = []
        #     a_set = self.postorderTraversal(root.left)
        #     b_set = self.postorderTraversal(root.right)
        #     if a_set:
        #         ret.extend(a_set)
        #     if b_set:
        #         ret.extend(b_set)
        #     ret.append(root.val)
        #     return ret