leetcode 144/145 Binary Tree Traversal (medium/hard)

144. Binary Tree Preorder Traversal (medium)
https://leetcode.com/problems/binary-tree-preorder-traversal/

题目描述
用迭代的方式前序遍历二叉树, 而不使用递归.

解题思路
前序遍历的顺序为 中->左->右, 考虑左边节点的始终比右边节点的要先遍历, 因此可以用栈, 先进后出的特性. 先将右边的节点入栈, 左边的节点后入栈, 这样左边的节点将先弹出.

代码

class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        res = []
        stack = [root]
        while stack:
            node = stack.pop()
            res.append(node.val)
            # 后加入的左节点将先出栈
            if node.right: stack.append(node.right)
            if node.left: stack.append(node.left)
        return res

145. Binary Tree Postorder Traversal (hard)
https://leetcode.com/problems/binary-tree-postorder-traversal/

题目描述
用迭代的方式后序遍历二叉树, 而不使用递归.

解题思路
后序遍历的顺序为 左->右->中, 我们发现其逆序列是 中->右->左, 可以参考144的前序遍历, 只不过此时是先遍历左边的节点, 然后才是左边的节点. 将最后得到的遍历序列逆序即可.

代码

class Solution:
    def postorderTraversal(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        res = []
        stack = [root]
        while stack:
            node = stack.pop()
            res.append(node.val)
            # 后加入的左节点将先出栈
            if node.left: stack.append(node.left)
            if node.right: stack.append(node.right)
        return res[::-1]