二叉树前中后序遍历统一模板

前序（根左右）

144. 二叉树的前序遍历

题目链接

Python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        ans = []  # 用于存储前序遍历结果的列表
        if not root:  # 如果根节点为空，直接返回空列表
            return ans
        stack = []  # 初始化一个栈，用于辅助遍历
        stack.append(root)  # 将根节点入栈
        while stack:  # 当栈不为空时，继续循环
            p = stack[-1]  # p指向栈顶元素，但不弹出
            if p:  # 如果栈顶元素不为空（即不是None）
                p = stack.pop()  # 将栈顶元素弹出并重新赋值给p
                # 将右子节点入栈（注意：此时不入处理，只是暂存）
                if p.right:
                    stack.append(p.right)
                # 将左子节点入栈（注意：此时也不处理，只是暂存）
                if p.left:
                    stack.append(p.left)
                # 将当前节点重新入栈，用于后续处理其值
                stack.append(p)
                # 插入一个None作为标记，表示下一次循环时需要处理该节点的值
                stack.append(None)
            else:  # 如果栈顶元素为空（即之前的None），表示需要处理其前一个节点的值
                stack.pop()  # 弹出这个None标记
                p = stack.pop()  # 弹出需要处理的节点
                ans.append(p.val)  # 将节点的值添加到结果列表中
        return ans  # 返回前序遍历的结果列表

JavaScript

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var preorderTraversal = function (root) {
    let res = []
    if (root===null) {
        return res
    }
    let st = []
    st.push(root)
    while (st.length > 0) {
        let p = st[st.length - 1]
        if (p !== null) {
            p = st.pop()
            if (p.right) {
                st.push(p.right)
            }
            if (p.left) {
                st.push(p.left)
            }
            st.push(p)
            st.push(null)
        } else {
            st.pop()
            p = st.pop()
            res.push(p.val)
        }
    }
    return res
};

---

中序（左根右）

94. 二叉树的中序遍历

题目链接

Python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        ans = []
        if not root:
            return ans
        stack = []
        stack.append(root)
        while stack:
            p = stack[-1]
            if p:
                p = stack.pop()
                # 右
                if p.right:
                    stack.append(p.right)
                # 根
                stack.append(p)
                stack.append(None)
                # 左
                if p.left:
                    stack.append(p.left)
            else:
                stack.pop()
                p = stack.pop()
                ans.append(p.val)
        return ans

JavaScript

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var inorderTraversal = function (root) {
    let res = []
    if (root === null) {
        return res
    }
    let st = []
    st.push(root)
    while (st.length > 0) {
        let p = st[st.length - 1]
        if (p !== null) {
            p = st.pop()
            if (p.right) {
                st.push(p.right)
            }
            st.push(p)
            st.push(null)
            if (p.left) {
                st.push(p.left)
            }
        } else {
            st.pop()
            p = st.pop()
            res.push(p.val)
        }
    }
    return res
};

---

后序（左右根）

145. 二叉树的后序遍历

题目链接

Python

# Definition for a binary tree node.
# 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: Optional[TreeNode]) -> List[int]:
        ans = []
        if not root:
            return ans
        stack = []
        stack.append(root)
        while stack:
            p = stack[-1]
            if p:
                # 根
                stack.append(None)
                # 右
                if p.right:
                    stack.append(p.right)
                # 左
                if p.left:
                    stack.append(p.left)
            else:
                stack.pop()
                p = stack.pop()
                ans.append(p.val)
        return ans

JavaScript

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var postorderTraversal = function (root) {
    let res = []
    if (root === null) {
        return res
    }
    let st = []
    st.push(root)
    while (st.length > 0) {
        let p = st[st.length - 1]
        if (p !== null) {
            st.push(null)
            if (p.right) {
                st.push(p.right)
            }
            if (p.left) {
                st.push(p.left)
            }
        } else {
            st.pop()
            p = st.pop()
            res.push(p.val)
        }
    }
    return res
};