栈---二叉树的前序遍历

最新推荐文章于 2024-02-29 17:31:53 发布

原创最新推荐文章于 2024-02-29 17:31:53 发布 · 835 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#栈 #二叉树 #前序遍历

算法专栏收录该内容

22 篇文章

订阅专栏

本文详细介绍了二叉树的前序遍历算法，提供了两种不同的栈实现方法，包括节点入栈及访问流程，适合初学者理解与实践。

题目
给定一个二叉树，返回它的前序遍历。
示例

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [1,2,3]

代码实现
方法一：
a.根节点入栈；
b.弹出根节点，访问根节点；判断右孩子是否为空，如果不为空，push入栈；判断左孩子是否为空，如果不为空，push入栈。
注：右孩子先入栈，左孩子后入栈，这样可以保证下一次出栈的时候，左孩子在栈顶。

 public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        if(root!=null){
            stack.push(root);
        }
        while (!stack.empty()){
            TreeNode top = stack.pop();
            list.add(top.val);
            if(top.right!=null){
                stack.push(top.right);
            }
            if(top.left!=null){
                stack.push(top.left);
            }
        }
        return list;
    }

方法二：
a.对于任意节点，将当前节点及其不为null的左孩子全部入栈并访问；
b.若当前栈顶元素无左孩子，弹出栈顶元素，然后将栈顶元素的右孩子置为当前节点；
c.继续执行a，b操作，直到当前节点为null并且栈为空则结束。

  public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        while (root!=null||!stack.empty()){
            while(root!=null){
                list.add(root.val);
                stack.push(root);
                root=root.left;
            }
            if(!stack.empty()){
                TreeNode top = stack.pop();
                root = top.right;
            }
        }
        return list;
    }