【leetcode】144.二叉树的前序遍历

题目

leetcode原题链接

给你二叉树的根节点 root ，返回它节点值的 前序 遍历。

示例 1：

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

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

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

思路

• 递归

  • 递归函数参数：当前节点node，用于存放结果的数组ans
  • 确定递归返回条件：当前节点为空，直接return
  • 单层递归逻辑：前序遍历是’根左右’，先存放中间节点的值，然后递归左节点，再递归右节点

• 迭代

  • 利用栈使用迭代的方法来完成前序遍历
  • 首先将中间节点入栈
  • 出栈一个节点，将该节点的右节点先入栈，然后再将左节点入栈。为什么右节点先入栈？这样出栈时才是左节点先出栈，右节点后出栈，也就是前序遍历的‘中左右’
  • 栈为空时就结束

代码

• 递归

/**
 * 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) {
    function traversal(node , ans){
        if(node === null) return

        // 放入中间节点的值
        ans.push(node.val)
        // 递归左右节点
        traversal(node.left , ans)
        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 preorderTraversal = function(root) {
    if(!root) return []
    let queue = []
    let ans = []
    queue.push(root)

    while(queue.length){
        const node = queue.pop()
        ans.push(node.val)
        if(node.right) queue.push(node.right)
        if(node.left) queue.push(node.left)
    }

    return ans
};

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