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

2024年6月10日 99次阅读
  1. 题目
    给定一个二叉树，返回它的前序遍历。
  2. 示例
输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [1,2,3]
  1. 代码实现
    方法一：
    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;
    }
点赞