LeetCode：144.二叉树的前序遍历 && 94.二叉树的中序遍历 && 145.二叉树的后序遍历

144.二叉树的前序遍历

题目

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

递归（三要素）

1. 确定递归函数的参数和返回值
2. 确定终止条件
3. 确定单层递归的逻辑

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

迭代

先将根节点放入栈中，然后将右孩子加入栈，再加入左孩子。出栈的时候就会是中左右的顺序。

// 前序遍历顺序：中-左-右，入栈顺序：中-右-左
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null) return result;
        Stack<TreeNode> stack = new Stack<>();
        //push：将root压入栈顶
        stack.push(root);
        while(!stack.isEmpty()){
        	//pop：删除栈顶对象，并返回该对象
            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;
    }
}

94.二叉树的中序遍历

题目

给定一个二叉树的根节点 root ，返回它的中序遍历 。

递归

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        inorder(root, result);
        return result;
    }
    public void inorder(TreeNode root, List<Integer> result){
        if(root == null) return;
        //左中右
        inorder(root.left, result);
        result.add(root.val);
        inorder(root.right, result);
    }
}

迭代

使用迭代法写中序遍历，借用指针的遍历来帮助访问节点，栈则用来处理节点上的元素

// 中序遍历顺序: 左-中-右 入栈顺序： 左-右
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null) return result;
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur= root;
        while(cur != null || !stack.isEmpty()){
            if(cur != null){
            	//pop：删除栈顶对象，并返回该对象
                stack.pop();
                cur = cur.left;
            }else{
                cur = stack.pop();
                result.add(cur.val);
                cur = cur.right;
            }
        }
        return result;
    }
}

145.二叉树的后序遍历

题目

给你一棵二叉树的根节点 root ，返回其节点值的后序遍历 。

递归

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        postorder(root, result);
        return result;
    }
    public void postorder(TreeNode root, List<Integer> result){
        if(root == null) return;
        //左右中
        postorder(root.left, result);
        postorder(root.right, result);
        result.add(root.val);
    }
}

迭代

后续遍历是左右中，只需要调整一下前序遍历的代码顺序，就变成中右左的遍历顺序，然后在反转result数组，输出的结果顺序就是左右中了。

// 后序遍历顺序 左-右-中 入栈顺序：中-左-右 出栈顺序：中-右-左， 最后翻转结果
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null) return result;
        Stack<TreeNode> stack = new Stack<>();
        //push：将root压入栈顶
        stack.push(root);
        while(!stack.isEmpty()){
        	//pop：删除栈顶对象，并返回该对象
            TreeNode node = stack.pop();
            result.add(node.val);
            if(node.left != null) stack.push(node.left);
            if(node.right != null) stack.push(node.right);
        }
        //反转
        Collections.reverse(result);
        return result;
    }
}