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

# 给定一个二叉树，返回它的 后序 遍历。 
# 
#  示例: 
# 
#  输入: [1,null,2,3]  
#    1
#     \
#      2
#     /
#    3 
# 
# 输出: [3,2,1] 
# 
#  进阶: 递归算法很简单，你可以通过迭代算法完成吗？ 
#  Related Topics 栈 树 
#   438  0


# 145.二叉树的后序遍历

# leetcode submit region begin(Prohibit modification and deletion)
# Definition for a binary tree node.
from collections import deque
from typing import List


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


'''
# 递归写法
class Solution:
    def postorderTraversal(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        return self.recur(root, [])

    def recur(self, root, ans):
        if not root:
            return
        self.recur(root.left, ans)
        self.recur(root.right, ans)
        ans.append(root.val)
        return ans
'''


class Solution:
    '''
    def postorderTraversal(self, root: TreeNode) -> List[int]:
        ans = []
        if not root:
            return ans
        tmp_deque = deque([root])
        while tmp_deque:
            cur = tmp_deque.pop()
            if cur.right:
                tmp_deque.append(cur.right)
            if cur.left:
                tmp_deque.append(cur.left)
            ans.append(cur.val)
        return ans
    # 错误的解: 我的妈呀，先序遍历都行，这就出错？
    '''

    '''
        为什么刚刚写的前序遍历的代码，不能和中序遍历通用呢，因为前序遍历的顺序是中左右，先访问的元素是中间节点，要处理的元素也是中间节点，所以刚刚才能写出相对简洁的代码，「因为要访问的元素和要处理的元素顺序是一致的，都是中间节点。」
    那么再看看中序遍历，中序遍历是左中右，先访问的是二叉树顶部的节点，然后一层一层向下访问，直到到达树左面的最底部，再开始处理节点（也就是在把节点的数值放进result数组中），这就造成了「处理顺序和访问顺序是不一致的。」
    那么「在使用迭代法写中序遍历，就需要借用指针的遍历来帮助访问节点，栈则用来处理节点上的元素。」

    def middleorderTraversal(self, root: TreeNode) -> List[int]:
        ans = []
        if not root:
            return ans
        cur = root
        tmp_deque = deque()
        while cur or tmp_deque:
            if cur:
                tmp_deque.append(cur)
                cur = cur.left
            else:
                cur = tmp_deque.pop()
                ans.append(cur.val)
                cur = cur.right
        return ans
    # 中序遍历 
    '''
    '''
        先根遍历然后反转
        根左右  反转  右左根     根右左 反转 左右根
    '''
    # leetcode submit region end(Prohibit modification and deletion)
    def postorderTraversal(self, root: TreeNode) -> List[int]:
        ans = []
        if not root:
            return ans
        tmp_stack = deque([root])
        while tmp_stack:
            cur = tmp_stack.pop()
            ans.append(cur.val)
            if cur.left:
                tmp_stack.append(cur.left)
            if cur.right:
                tmp_stack.append(cur.right)
        return ans[::-1]
# https://mp.weixin.qq.com/s/WKg0Ty1_3SZkztpHubZPRg

if __name__ == '__main__':
    sol = Solution()
    a, b, c, d, e = TreeNode(0), TreeNode(1), TreeNode(2), TreeNode(3), TreeNode(4)
    # a.left = b
    # b.left = c
    # b.right = d
    # a.right = e
    b.right = c
    c.left = d

    sol = Solution()
    print(sol.postorderTraversal(b))
    # print(sol.postorderTraversal([]))