二叉树的五种遍历方式

目录

1、前序遍历

(1)递归实现前序遍历

(2)非递归实现前序遍历

 2、中序遍历

(1)递归实现中序遍历

(2)非递归实现中序遍历

 3、后序遍历

(1)递归实现后序遍历

(2)非递归实现后序遍历

4、层序遍历

5、之字形遍历


二叉树是一种重要的数据结构,其遍历方式分为:深度遍历和广度遍历,深度遍历有前序、中序以及后序三种遍历方法,广度遍历即就是层次遍历。如下图:

二叉树的五种遍历方式_第1张图片

class TreeNode{
    int val;
    TreeNode left;
    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;
    }
}

1、前序遍历

遍历顺序:根节点---左子树---右子树

如上图,遍历结果应该为:12457836

(1)递归实现前序遍历

    /**
     * 递归实现前序遍历
     * @param treeNode  树的根节点
     */
    public static void preOrder1(TreeNode treeNode){
        // 若根节点为空,直接返回
        if(treeNode == null){
            return;
        }
        //打印根节点
        System.out.print(treeNode.val + "\t");
        // 遍历根节点的左子树
        preOrder1(treeNode.left);
        // 遍历根节点的右子树
        preOrder1(treeNode.right);
    }

(2)非递归实现前序遍历

非递归实现可以通过辅助栈或者辅助队列实现。以下代码为辅助栈的实现方式:

    /**
     * 非递归实现前序遍历
     * @param treeNode  根节点
     */
    public static void preOrder2(TreeNode treeNode){
        // 如果根节点为空,直接返回。
        if(treeNode == null){
            return;
        }
        // 辅助栈
        Stack stack = new Stack<>();
        // 根节点入栈
        stack.push(treeNode);
        // 当栈不为空
        while(!stack.isEmpty()){
            //取出栈顶元素
            TreeNode node = stack.pop();
            //打印根节点
            System.out.print(node.val + "\t");
            // 如果使用的是辅助栈,则先将根节点的右子节点入栈;如果是辅助队列,则先将根节点的左子节点入队列。因为栈是先进后出,队列是先进入=先出
            if(node.right != null){
                stack.push(node.right);
            }
            // 根节点的右子节点入栈
            if(node.left != null){
                stack.push(node.left);
            }
        }
    }

二叉树的五种遍历方式_第2张图片

 2、中序遍历

遍历顺序:左子树---根节点---右子树

如上图,遍历结果应该为:42758136

(1)递归实现中序遍历

    /**
     * 递归实现中序遍历
     * @param treeNode  树的根节点
     */
    public static void inOrder1(TreeNode treeNode){
        // 若根节点为空,直接返回
        if(treeNode == null){
            return;
        }
        // 遍历根节点的左子树
        inOrder1(treeNode.left);
        //打印根节点
        System.out.print(treeNode.val + "\t");
        // 遍历根节点的右子树
        inOrder1(treeNode.right);
    }

(2)非递归实现中序遍历

    /**
     * 非递归实现中序遍历
     * @param treeNode  根节点
     */
    public static void inOrder2(TreeNode treeNode){
        // 如果根节点为空,直接返回。
        if(treeNode == null){
            return;
        }
        // 辅助栈
        Stack stack = new Stack<>();
        // 临时指针
        TreeNode cur = treeNode;
        // 当栈不为空
        while(cur != null || !stack.isEmpty()){
            // 左节点入栈
            while(cur != null){
                stack.push(cur);
                cur = cur.left;
            }
            //取出栈顶元素
            cur = stack.pop();
            //打印左节点
            System.out.print(cur.val + "\t");
            // 指向右节点
            cur = cur.right;
        }
    }

 3、后序遍历

遍历顺序:左子树---右子树---根节点

如上图,遍历结果应该为:47852631

(1)递归实现后序遍历

    /**
     * 递归实现后序遍历
     * @param treeNode  树的根节点
     */
    public static void postOrder1(TreeNode treeNode){
        // 若根节点为空,直接返回
        if(treeNode == null){
            return;
        }
        // 遍历根节点的左子树
        postOrder1(treeNode.left);

        // 遍历根节点的右子树
        postOrder1(treeNode.right);
        //打印根节点
        System.out.print(treeNode.val + "\t");
    }

(2)非递归实现后序遍历

 

    /**
     * 非递归实现后序遍历
     * @param treeNode  根节点
     */
    public static void postOrder2(TreeNode treeNode){
        // 如果根节点为空,直接返回。
        if(treeNode == null){
            return;
        }
        // 辅助栈
        Stack stack = new Stack<>();
        // 临时指针
        TreeNode cur = treeNode, pre = null;
        // 当栈不为空
        while(cur != null || !stack.isEmpty()){
            // 左节点入栈
            while(cur != null){
                stack.push(cur);
                cur = cur.left;
            }
            //取出栈顶元素
            cur = stack.get(stack.size()-1);
            if(cur.right == null || pre == cur.right){
                stack.pop();
                System.out.print(cur.val + "\t");
                pre = cur;
                cur = null;
            }else{
                // 指向右节点
                cur = cur.right;
            }
        }
    }

4、层序遍历

遍历顺序:逐层遍历

如上图,遍历结果应该为:12345678。通过辅助队列,在取出节点的同时,将当前节点的左右节点分别入队。

    /**
     * 层序遍历
     * @param treeNode
     */
    public static void levelOrder(TreeNode treeNode){
        //根节点为空,直接返回
        if(treeNode == null){
            return;
        }
        //辅助队列
        Queue queue = new LinkedList<>();
        //根节点入队列
        queue.offer(treeNode);
        //当栈不为空
        while(!queue.isEmpty()){
            //取出队首元素
            TreeNode node = queue.poll();
            System.out.print(node.val + "\t");
            //将节点的左节点入队
            if(node.left != null){
                queue.offer(node.left);
            }
            //节点的右节点入队
            if(node.right != null){
                queue.offer(node.right);
            }
        }
    }

 

5、之字形遍历

这是牛客上的一道题。这个之字形遍历也可理解为Z字形遍历,以上树为例,其遍历结果为:1,3,2,4,5,6,8,7。本质还是二叉树的层序遍历,只不过在便利的时候,要将偶数层的节点逆序。

代码:

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
    public ArrayList > Print(TreeNode root) {
        //存储结果
        ArrayList> result = new ArrayList<>();
        //如果根节点为空,则直接返回
        if(root == null){
            return result;
        }
        //辅助队列
        Queue queue = new LinkedList<>();
        //根节点入队
        queue.offer(root);
        //是否转向
        boolean flag = false;
        while(!queue.isEmpty()){
            //获取队列长度
            int size = queue.size();
            //存储每一层的遍历结果
            ArrayList list = new ArrayList<>();
            for(int i=0; i < size; i++){
                //取出队列元素
                TreeNode node = queue.poll();
                if(node == null){
                    continue;
                }
                if(!flag){
                    list.add(node.val);
                }else{
                    list.add(0, node.val);
                }
                //左右节点各入队
                queue.offer(node.left);
                queue.offer(node.right);
            }
            //如果有值,存入结果集
            if(list.size() > 0){
                result.add(list);
            }
            //转向
            flag = !flag;
        }
        return result;
    }

}

你可能感兴趣的:(Java,java,数据结构,算法)