Java二叉树的遍历（先序、中序、层序、后序）

先序遍历

• 求给定的二叉树的前序遍历。

思想

• 输出顺序
  根>左>右

算法（非递归）

• 使用栈来实现

    public ArrayList<Integer> preorderTraversal(TreeNode root) {
        //结果的存储
        ArrayList<Integer> result = new ArrayList<>();
        if (root == null) return result;
        //使用栈来实现
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        //循环取数据，直到栈为空
        while (!stack.isEmpty()) {
            TreeNode treeNode = stack.pop();
            result.add(treeNode.val);
            //先右结点入栈
            if (treeNode.right != null) stack.push(treeNode.right);
            //再左节点入栈
            if (treeNode.left != null) stack.push(treeNode.left);
        }
        return result;
    }

算法（递归）

    public ArrayList<Integer> preorderTraversal(TreeNode root) {
        //结果存储
        ArrayList<Integer> result = new ArrayList<>();
        //判空
        if (root == null) return result;
        //先根遍历，先存储根结点
        result.add(root.val);
        //递归，左子树
        result.addAll(preorderTraversal(root.left));
        //递归，右子树
        result.addAll(preorderTraversal(root.right));
        return result;
    }

中序遍历

• 给出一棵二叉树，返回这棵树的中序遍历

思想

• 输出顺序
  左>中>右

算法（非递归）

    public ArrayList<Integer> inorderTraversal(TreeNode root) {
        //存储结果
        ArrayList<Integer> result = new ArrayList<>();
        //判空
        if (root == null) return result;
        TreeNode temp = root;
        //使用栈来解决
        Stack<TreeNode> stack = new Stack<>();
        while (!stack.isEmpty() || temp != null) {
            //先将所有左子树添加到栈中
            while (temp != null) {
                stack.push(temp);
                temp = temp.left;
            }
            //出栈
            temp = stack.pop();
            result.add(temp.val);
            //添加右结点入栈
            temp = temp.right;
        }
        return result;
    }

算法（递归）

    public ArrayList<Integer> inorderTraversal(TreeNode root) {
        //存储结果
        ArrayList<Integer> result = new ArrayList<>();
        //判空
        if (root == null) return result;
        //递归依次调用，左中右
        result.addAll(inorderTraversal(root.left));
        result.add(root.val);
        result.addAll(inorderTraversal(root.right));
        return result;
    }

层序遍历

• 牛客网，求二叉树的层序遍历
• LeetCode，求二叉树的层序遍历

思想

• 1.根结点先入队
• 2.出队访然后问其左右孩子结点并入队
• 3.重复2的步骤直到队列为空

算法

    public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
        if (root == null) return new ArrayList<>();
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            ArrayList<Integer> temp = new ArrayList<>();
            //记录当前queue的大小，这几个值属于同一层
            int count = queue.size();
            for (int i = 0; i < count; i++) {
                //得到当前遍历结点
                TreeNode treeNode = queue.poll();
                temp.add(treeNode.val);
                //遍历子结点，添加到队列中
                if (treeNode.left != null) queue.offer(treeNode.left);
                if (treeNode.right != null) queue.offer(treeNode.right);
            }
            result.add(temp);
        }
        return result;
    }
}

后序遍历

• 用递归的方法对给定的二叉树进行后序遍历。

思想

• 输出顺序（后根遍历）
  左，右，根

算法（递归）

    public ArrayList<Integer> postorderTraversal(TreeNode root) {
        //存储结果
        ArrayList<Integer> result = new ArrayList<>();
        //判空
        if (root == null) return result;
        //依次递归，左、右、根
        result.addAll(postorderTraversal(root.left));
        result.addAll(postorderTraversal(root.right));
        result.add(root.val);
        return result;
    }