Java数据结构——二叉树的遍历

首先,什么是二叉树的遍历呢?

二叉树的遍历是指从根结点出发,按照某种次序依次访问二叉树中所有结点,使得每个结点被访问一次且仅被访问一次。

 为什么要遍历二叉树呢?

因为计算机只会处理线性序列,而我们研究遍历,就是把树中的结点变成某种意义的线性序列,这给程序的实现带来了好处。

 在遍历二叉树之前,我们首先要创建一个二叉树,想要创建如下图的二叉树,就要先进行二叉树的扩展,也就是将二叉树每个节点的空指针引出一个虚节点,其值为一个特定值,比如'#' 。处理后的二叉树称为原二叉树的扩展二叉树,扩展二叉树的每个遍历序列可以唯一确定一棵二叉树,我们采用前序遍历创建二叉树。前序遍历序列:124##5##36##7##。

mark        mark

定义内部节点类

class TreeNode {
        char value; //数据
        TreeNode left; //左子树
        TreeNode right; //右子树
        public TreeNode(char value){
            this.value=value;
        }
    }

根据前序遍历创建二叉树

 //根据字符串创建二叉树
    public int i=0;  
    TreeNode createTree(String s){
        TreeNode root=null;
        if(s.charAt(i)!='#'){
            root=new TreeNode(s.charAt(i));
            i++;
            root.left=createTree(s);
            root.right=createTree(s);
        }else{
            i++;
        }
        return root;
    }

二叉树的四种遍历方法

  1. 前序遍历:先访问根节点,然后前序遍历左子树,再前序遍历右子树
  2. 中序遍历:中序遍历根节点的左子树,然后访问根节点,最后遍历右子树
  3. 后序遍历:后序遍历根节点的左子树,然后后序遍历根节点的右子树,最后访问根节点
  4. 层序遍历:从根节点开始,从上往下逐层遍历,在同一层,按从左到右的顺序对节点逐个访问

1. 二叉树的前序遍历

                                                        mark

递归实现二叉树的前序遍历 

具体步骤:

  1. 先访问根节点
  2. 再递归遍历左子树
  3. 最后递归遍历右子树
    //二叉树前序递归遍历
    void binaryTreePrevOrder(TreeNode root){
          if(root==null){
              return;
          }
        System.out.print(root.value+" ");
        binaryTreePrevOrder(root.left);
        binaryTreePrevOrder(root.right);
    }

非递归实现二叉树的前序遍历 

具体步骤:

  1. 首先申请一个新的栈,记为stack;
  2. 将根节点root压入stack中;
  3. 每次从stack中弹出栈顶节点,记为cur,然后打印cur值;
  4. 如果cur右孩子不为空,则将右孩子压入栈中;如果cur的左孩子不为空,将其压入stack中;
  5. 重复步骤3,直到stack为空.
//二叉树前序非递归遍历   
 public static void PreOrder(TreeNode root) {
        if (root == null) {
            return;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!(stack.empty())) {
            TreeNode cur = stack.pop();
            System.out.print(cur.value+" ");
            if (cur.right != null) {
                stack.push(cur.right);
            }
            if (cur.left != null) {
                stack.push(cur.left);
            }
        }
    }

2. 二叉树的中序遍历

                                                 mark                              

递归实现二叉树的中序遍历

具体步骤:

  1. 先中序遍历左子树
  2. 再访问根节点
  3. 最后中序遍历右子树
//递归中序遍历二叉树  
  void binaryTreeInOrder(TreeNode root){
       if(root==null){
           return;
       }
       binaryTreeInOrder(root.left);
       System.out.print(root.value+" ");
       binaryTreeInOrder(root.right);
    }

 非递归实现二叉树的中序遍历

具体步骤:

  1. 申请一个栈,记为stack,申请一个变量cur,初始时cur为根节点root;
  2. 先把cur节点压入栈中,对以cur节点为头的整棵子树来说,依次把整棵树的左子树压入栈中,即不断令cur=cur.left,然后重复步骤2;
  3. 不断重复步骤2,直到发现cur为空,此时从stack中弹出一个节点记为node,打印node的值,并让cur = node.right,然后继续重复步骤2;
  4. 当stack为空并且cur为空时结束。
    //二叉树的中序遍历非递归
    void binaryTreeInOrderNonR(TreeNode root){
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;
        TreeNode node = null;
        while (cur != null || !stack.empty()) {
            while (cur != null) {
                stack.push(cur);
                cur = cur.left;
            }
            node = stack.pop();
            System.out.print(node.value+" ");
            cur = node.right;
        }
    }

3. 二叉树的后序遍历

                                                     mark

  递归实现二叉树的后序遍历

具体步骤:

  1. 先后序遍历左子树
  2. 再后序遍历右子树
  3. 最后访问根节点
     //后序递归遍历二叉树
    void binaryTreeLastOrder(TreeNode root){
        if(root==null){
            return;
        }
        binaryTreeLastOrder(root.left);
        binaryTreeLastOrder(root.right);
        System.out.print(root.value+" ");
    }

 非递归实现二叉树的后序遍历

具体步骤:

  1. 申请一个栈stack,将头节点压入stack,同时设置两个变量 h 和 c,在整个流程中,h代表最近一次弹出并打印的节点,c代表当前stack的栈顶节点,初始时令h为头节点,,c为null;
  2. 每次令c等于当前stack的栈顶节点,但是不从stack中弹出节点,此时分一下三种情况:

        (1)如果c的左孩子不为空,并且h不等于c的左孩子,也不等于c的右孩子,则吧c的左孩子压入stack中

       (2)如果情况1不成立,并且c的右孩子不为空,并且h不等于c的右孩子,则把c的右孩子压入stack中;

        (3)如果情况1和2不成立,则从stack中弹出c并打印,然后令h等于c;

    3. 一直重复步骤2,直到stack为空.

    //二叉树的后序遍历非递归
    void binaryTreePostOrderNonR(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;
        TreeNode prev = null;
        while (cur != null || !stack.empty()) {
            while (cur != null) {
                stack.push(cur);
                cur = cur.left;
            }
            //左边都入栈
            cur = stack.peek();//最左子树
            //cur.right == prev 代表的是 cur的右边已经打印过了
            if(cur.right == null || cur.right == prev) {
                stack.pop();
                System.out.print(cur.value+" ");
                prev = cur;
                cur = null;
            }else {
                cur = cur.right;
            }
        }
    }

  4. 二叉树的层序遍历

                                                       mark

 具体步骤:

  1. 首先申请一个新的队列,记为queue;
  2. 将根节点root放入queue中;
  3. 每次从queue中出队,记为cur,然后打印cur的值,如果cur左孩子不为空,则将左孩子入队;如果cur的右孩子不为空,则将右孩子入队;
  4. 重复步骤3,直到queue为空。
    //二叉树的层序遍历
    void binaryTreeLevelOrder(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        if(root != null) {
            //入队
            queue.offer(root);
        }
        while (!queue.isEmpty()) {
            //1、拿到队头的元素 把队头元素的左右子树入队
            TreeNode cur = queue.poll();
            System.out.print(cur.value+" ");
            //2、不为空的时候才能入队
            if(cur.left != null) {
                queue.offer(cur.left);
            }
            if(cur.right != null) {
                queue.offer(cur.right);
            }
        }
    }

总结:

  1. 前序,中序,后序遍历都要先访问左子树的左子树的左子树...(cur = cur.left)直到cur.left=null为止
  2. 后序遍历难点:如何知道右子树有没有打印
  3. 一般情况下,元素出栈之前要打印
发布了50 篇原创文章 · 获赞 39 · 访问量 8267

猜你喜欢

转载自blog.csdn.net/Racheil/article/details/97006319