LeetCode—144. 二叉树的前序遍历

144 . 二叉树的前序遍历

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

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

思路一：递归
用递归的方式很简单，就是遇到根节点就增加到数组中，然后左子树不为空就递归遍历左子树，右子树不为空就递归遍历右子树。

class Solution {
    public void preorder(TreeNode root,List<Integer> ans){
        if(root == null) return;
        ans.add(root.val);
        preorder(root.left,ans);
        preorder(root.right,ans);
    }
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        preorder(root,ans);
        return ans;
    }
}

---

思路二：迭代
创建一个栈，先压入根节点，然后判断右子树是否为空，不为空的话压入右子树。继续判断左子树是否为空，不为空的压入左子树。
为什么先压入右子树，不先压入左子树呢？因为栈是先入后出的呀。

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

---

思路三：迭代（使用两个栈）
创建两个栈，一个存放节点的栈，一个状态栈，状态栈表示当前压入的栈是什么节点。
0表示左节点，1表示右节点，2表示根节点。
首先压入根节点，此时状态栈压入2。
然后判断左子树是否为空，不为空的话，就进入左子树，状态为0。而如果当前的左节点是当前左子树的根节点，状态为2。
只有当右子树为空时，就会弹栈。

public List<Integer> preorderTraversal(TreeNode root) {
    List<Integer> result = new ArrayList<>();
    if(root == null) return result;
    Stack<TreeNode> stack = new Stack<>();      //压入根节点的栈
    Stack<Integer> statusStack = new Stack<>(); //状态栈，表示当前压入的是哪个节点
    //压入根节点
    stack.push(root);
    statusStack.push(2);
    while(!stack.isEmpty()){
        switch(statusStack.pop()){
                //0表示左节点，1表示右节点，2表示根节点
            case 0:{
                statusStack.push(1);
                if(stack.peek().right != null){
                    stack.push(stack.peek().right);
                    statusStack.push(2);
                }
                break;
            }
            case 1:{
                stack.pop();
                break;
            }
            case 2:{
                result.add(stack.peek().val);
                statusStack.push(0);
                if(stack.peek().left != null){
                    stack.push(stack.peek().left);
                    statusStack.push(2);
                }
                break;
            }
        }
    }
    return result;
}

第三种思路优点绕，为了弄懂它，就针对它花了一个图，画着画着把自己绕进去了，没画完。