【leetcode】94.二叉树的中序遍历

题目

leetcode原题链接

给定一个二叉树的根节点 root ，返回 它的 中序 遍历 。

示例 1：

输入：root = [1,null,2,3]
输出：[1,3,2]

示例 2：
输入：root = []
输出：[]

示例 3：
输入：root = [1]
输出：[1]

提示：
树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100

思路

• 递归
  • 中序遍历的顺序是“左中右”
  • 递归函数的参数是当前节点和结果数组
  • 返回条件是当前所遍历的节点为空，直接返回
  • 单层递归逻辑是先递归左节点，再将中间节点的值放入结果数组，最后递归右节点。
• 迭代
  • 由于我们访问二叉树是从根节点，也就是中间节点开始，但是中序遍历要求先遍历左节点，因此我们需要记录中间节点，并先去访问它的左子树。
  • 我们使用栈来记录节点
  • 一：只要当前遍历到的节点不为空，就要记录该节点并且去访问该节点的左节点。
  • 二：前节点为空时，说明左节点为空，需要将中间节点的值加入结果数组，此时的中间节点位于栈顶，将中间节点出栈，将它的值加入结果数组。
  • 三：接下来访问右节点，右节点不为空，回到第一步，右节点为空，回到第二步。
  • 当栈为空时结束遍历。

代码

• 递归

/**
 * 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) {

    function traversal(node , ans){
        if(!node) return 

        // 先递归左节点
        traversal(node.left , ans)
        // 将中间节点的值放入结果数组
        ans.push(node.val)
        // 递归右节点
        traversal(node.right , ans)
    }

    let ans = []
    traversal(root , ans)
    return ans
};

• 迭代

/**
 * 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 stack = []
    let ans = []
    // cur为当前节点
    let cur = root
    while(cur || stack.length){
        if(cur){
            stack.push(cur)
            // 遍历左节点
            cur = cur.left
        }else{  //当前节点为空，要处理栈顶（中间）节点
            const node = stack.pop()
            ans.push(node.val)
            // 中间节点遍历完，遍历右子树
            cur = node.right
        }
    }

    return ans
};

关注我的专栏，每天更新三道leetcode题解，一起变强！