二叉树遍历详解

二叉树的遍历方式是最基本，也是最重要的一类题目，我们将从「前序」、「中序」、「后序」、「层序」四种遍历方式出发，总结他们的递归和迭代解法。

一、二叉树定义
      二叉树（Binary tree）是树形结构的一个重要类型。许多实际问题抽象出来的数据结构往往是二叉树形式，即使是一般的树也能简单地转换为二叉树，而且二叉树的存储结构及其算法都较为简单，因此二叉树显得特别重要。二叉树特点是每个结点最多只能有两棵子树，且有左右之分。简单来说，就是一个包含节点，以及它的左右孩子的一种数据结构

假设二叉树的节点定义如下

public class TreeNode {
    public int val;
    public TreeNode left;
    public TreeNode right;

    public TreeNode() {
    }

    public TreeNode(int val) {
        this.val = val;
    }

    public TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

二、层序遍历

层序遍历比较简单，按照从上到下，从左到右的顺序逐次遍历。此处借用队列的先入先出特性来实现，具体代码如下

public static void levelTraverse(TreeNode root) {
        if (root == null) {
            return;
        }
        Queue<TreeNode> queue = new LinkedList<>();

        //初始化时把root放入队列
        queue.offer(root);

        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();

            //打印节点的值
            System.out.print(node.val + " ");

            //队列是先入先出，所以此处先遍历左节点
            if (node.left != null) {
                queue.offer(node.left);
            }

            if (node.right != null) {
                queue.offer(node.right);
            }
        }
    }

三、前序遍历（根节点，左子树，右子树）

1、递归实现：二叉树遍历的递归形式比较容易实现，直接按照根节点，左子树，右子树的顺序 逐次遍历即可

private static void preOrderTraversal0(TreeNode root) {
        if (root == null) {
            return;
        }

        //打印根节点
        System.out.print(root.val + " ");

        //打印左节点
        preOrderTraversal0(root.left);

        //打印右节点
        preOrderTraversal0(root.right);
    }

2、迭代实现

2.1 迭代解法一

过程如下：

• 初始化栈，并将根节点入栈；

• 当栈不为空时：
  弹出栈顶元素 node，并将值添加到结果中；
  如果 node 的右子树非空，将右子树入栈；
  如果 node 的左子树非空，将左子树入栈；
  由于栈是“先进后出”的顺序，所以入栈时先将右子树入栈，这样使得前序遍历结果为 “根->左->右”的顺序。

经过上面图的讲解，代码就比较简单，代码如下：

private static void preOrderTraversal1(TreeNode root) {
        if (null == root) {
            return;
        }

        //定义一个栈方便后续遍历
        Stack<TreeNode> stack = new Stack<>();

        //初始化
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();

            //每一次出栈都打印节点的值
            System.out.print(node.val + " ");

            //栈是先进后出的，所以先处理右子树入栈，再左子树入栈
            if (node.right != null) {
                stack.push(node.right);
            }
            if (node.left != null) {
                stack.push(node.left);
            }
        }
    }

2.2 迭代解法二

（1）思路稍有不同，先定义一个节点cur指向root节点，先将cur节点和所有的左孩子入栈同时打印出cur节点的值，直至 cur 为空，用一个 while 循环实现。

（2）随后出栈一个节点，定义为node，执行 cur = node.right，随后继续执行 操作（1）

经过上面分析，代码中的（1）和（2）分别对应上述的描述，代码如下：

private static void preOrderTraversal2(TreeNode root) {
        if (root == null) {
            return;
        }
        Stack<TreeNode> stack = new Stack<>();
        //定义一个cur指向root
        TreeNode cur = root;
        while (!stack.isEmpty() || cur != null) {
            //（1）只要cur！=null，则打印值，同时cur入栈，同时设置cur = cur.left
            while (cur != null) {
                System.out.print(cur.val + " ");
                stack.push(cur);
                cur = cur.left;
            }

            //（2）如果cur == null,则出栈一个节点，同时设置cur = node.right,同时继续执行（1）
            TreeNode node = stack.pop();
            cur = node.right;
        }
    }

四、中序遍历（左子树，根节点，右子树）

1、递归实现：二叉树遍历的递归形式比较容易实现，直接按照（左子树，根节点，右子树）的顺序 逐次遍历即可

//中序遍历
    private static void inOrderTraversal0(TreeNode root) {
        if (root == null) {
            return;
        }

        //打印左节点
        inOrderTraversal0(root.left);

        //打印根节点
        System.out.print(root.val + " ");

        //打印右节点
        inOrderTraversal0(root.right);
    }

2、迭代解法：（左子树，根节点，右子树）

（1）与前序遍历的逻辑差不多，前序遍历是入栈的时候打印值，但是中序遍历是先处理左节点，再处理根节点，最后遍历右节点，所以遍历时不打印值，出栈时打印值，先定义一个节点cur指向root节点，先将cur节点和所有的左孩子入栈 直至 cur 为空，用一个 while 循环实现。

（2）随后出栈一个节点，定义为node，打印节点的值，执行 cur = node.right，随后继续执行 操作（1）

经过上面处理后 root 节点的左子树处理完毕，接下来继续处理右子树，也是重复的过程，经过上面分析，代码如下：

private static void inOrderTraversal1(TreeNode root) {
        if (root == null) {
            return;
        }
        TreeNode cur = root;
        Stack<TreeNode> stack = new Stack<>();
        while (!stack.isEmpty() || cur != null) {
            //（1）如果cur不等于空，一直入栈，同时执行cur = cur.left,目的是找到最左节点
            while (cur != null) {
                stack.push(cur);
                cur = cur.left;
            }

            //(2)如果cur为空，则出栈一个元素，同时打印值，接下来处理右子树，右子树也是调用（1）同步处理
            TreeNode node = stack.pop();
            System.out.print(node.val + " ");
            cur = node.right;
        }
    }

五、后续遍历（左子树，右子树，根节点）

1、递归实现：二叉树遍历的递归形式比较容易实现，直接按照（左子树，右子树，根节点）的顺序 逐次遍历即可

private static void postOrderTraversal0(TreeNode root) {
        if (root == null) {
            return;
        }

        //打印左节点
        postOrderTraversal0(root.left);

        //打印右节点
        postOrderTraversal0(root.right);

        //打印根节点
        System.out.print(root.val + " ");

    }

2、迭代实现：（二叉树的后续遍历，先左子树，右子树，最后根结点），可以定义两个辅助栈，一个栈用于辅助遍历，一个栈用于存放结果，从root开始遍历时先遍历到跟结点，但是根节点又需要最后输出，所以可以借助栈的（先进后出的）特性实现先进入的节点最后输出

经过上面的图解代码如下：

public static void postOrderTraversal(TreeNode root) {
        if (root == null) {
            return;
        }
        Stack<TreeNode> stack1 = new Stack<TreeNode>();

        //起始时root入栈
        stack1.push(root);

        //定义一个result栈
        Stack<TreeNode> result = new Stack<TreeNode>();
        while (!stack1.isEmpty()) {
            TreeNode node = stack1.pop();
            result.push(node);
            //先左节点入栈
            if (node.left != null) {
                stack1.push(node.left);
            }
            //再右节点入栈
            if (node.right != null) {
                stack1.push(node.right);
            }
        }

        //最后result栈中依次出栈即为结果
        while (!result.isEmpty()) {
            System.out.print(result.pop().val + " ");
        }
    }

上面仅记录个人的理解。有错误麻烦指正，感谢。