[路飞]_leetcode-144-二叉树的前序遍历-迭代算法

给你二叉树的根节点 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

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

针对本题之前写过一篇题解文章，因为题目过于简单，没有完整读题，导致没注意到进阶要求，今天主要讲解下如果通过迭代算法实现二叉树的前序遍历。

首先我们来看一下前序遍历的过程：

leetcode-144-二叉树的前序遍历-遍历过程.gif

可以看到，如果当前节点有左子树，会优先处理左子树，直到叶子节点位置，然后会处理与之对应的右子树

如果右子树有左子树，依然优先处理左子树，然后处理与之对应的右子树

然后继续向上回溯处理父节点的右子树，直到回溯到根节点，处理根节点的右子树

所以，我们可以从根节点开始迭代，将根节点的值放入到结果数组中，然后将根节点入栈，然后向下处理当前节点的左子树，直到叶子节点位置

此时，栈顶保存的就是当前叶子节点，将栈顶元素弹出，当前节点更新为栈顶元素的右子树，因为该节点为叶子节点，所以右子树为空，此时继续弹出栈顶元素，此时栈顶元素为叶子节点的父节点，将当前元素更新为它的右子树

右子树处理到最后依然会来到叶子节点，此时栈顶元素为该叶子节点，弹出该节点，因为该节点为叶子节点，所以右子树为空，继续弹出栈顶元素，则继续向上回溯

整个遍历过程一直重复这样的过程，直到处理完整棵二叉树的最右侧节点，此时栈为空且当前节点为空，结束循环，完成二叉树的前序遍历。

动画演示如下：

leetcode-144-二叉树的前序遍历-迭代过程.gif

代码如下：

var preorderTraversal = function(root) {
  // 特判如果是空树，返回空数组
  if(root===null) return [];
  // 初始化结果数组 栈
  const res = [],stack = [];
  // 当当前节点不为空或者栈不为空的时候，遍历二叉树
  while(root!==null || stack.length){
    // 如果当前节点不为空，处理它的左子树
    while(root!==null){
      res.push(root.val);
      stack.push(root);
      root = root.left;
    }
    // 如果当前节点没有左子树，向上回溯处理父节点的右子树
    root = stack.pop().right;
  }
  // 返回结果数组
  return res;
};
复制代码

至此我们就完成了 leetcode-144-二叉树的前序遍历-迭代算法

如有任何问题或建议，欢迎留言讨论！

[路飞]_leetcode-144-二叉树的前序遍历-迭代算法

猜你喜欢