144. 二叉树的前序遍历简单树递归迭代

Posted 2023-02-28 pre_eminent

题目

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

---

示例 1：

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

---

示例 2：

输入：root = []
输出：[]

---

示例 3：

输入：root = [1]
输出：[1]

---

示例 4：

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

---

示例 5：

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

---

提示：

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

---

进阶：递归算法很简单，你可以通过迭代算法完成吗？

---

思路：

1. 递归

2. 迭代 （通过栈）视频教程

3. Morris 遍历 （todo）

---

解法：

/**
 * 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) 
    // result不能定为全局变量，会污染其他用例
    const result = [];
    // dfs(root, result);
    iterate(root, result);
    return result;
;

// 解法2：迭代（栈）
function iterate(root, result) 
    // 根结点入栈
    const stack = [];
    stack.push(root)

    // 只要栈不为空
    while(stack.length) 
        // 取出 栈顶 结点
        const top = stack.pop();
        // 关键
        if (top) 
            result.push(top.val);
         else 
            // continue会让外层循环去判断
            continue;
        
        // 处理top结点的子节点
        // 因为是前序遍历：根左右
        // 所以：把栈顶子结点 先右后左顺序入栈
        // 这样出栈时，就是先左后右了
        stack.push(top.right);
        stack.push(top.left);
    


// 解法1：递归
function dfs(root, result) 
    if (!root) 
        return;
    
    result.push(root.val);
    dfs(root.left, result);
    dfs(root.right, result);