二叉树的遍历

2023-04-17  本文已影响0人  碎念枫子

是一种经常用到的结构,用来模拟具有树状结构性质的数据集合

的每一个节点有一个和一个包含所有子节点的列表,从图的观点来看,树也可视为一个拥有N 个节点N-1 条边的一个有向无环图

二叉树是一种更为典型的树状结构。如它名字所描述的那样,二叉树是每个节点最多有两个子树的树结构,通常子树被称作“左子树”“右子树”

树的遍历

首先访问根节点,然后遍历左子树,最后遍历右子树

      F
    /   \
   B      G
  / \       \
 A   D       I
    / \     / 
  C    E   H
//遍历结构是 
F、B、A、D、C、E、G、I、H

先遍历左子树,然后访问根节点,最后访问右子树

      F
    /   \
   B      G
  / \       \
 A   D       I
    / \     / 
  C    E   H
//遍历结构是 
A、B、C、D、E、F、G、H、I

先遍历左子树,然后遍历右子树,最后访问根节点

      F
    /   \
   B      G
  / \       \
 A   D       I
    / \     / 
  C    E   H
//遍历结构是 
A、C、E、D、B、H、I、G、F

总结

所谓前中后都是根据根节点的位置来命名的。后序在数学表达式中广泛使用,编写程序来解析后缀法更加容易。

这里举一个例子:



对于这个图,我们使用中序遍历很容易能找出表达式:4x(7-2)+5

如果你想对这棵树进行后序遍历,使用栈来处理表达式会变得更加容易。 每遇到一个操作符,就可以从栈中弹出栈顶的两个元素,计算并将结果返回到栈中。

遍历

我们有三种常用的遍历方法、递归、迭代、morris,我们以给定的节点类如下,分别介绍三种遍历方法

 public class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode() {}
      TreeNode(int val) { this.val = val; }
      TreeNode(int val, TreeNode left, TreeNode right) {
          this.val = val;
          this.left = left;
          this.right = right;
      }
  }
```java
public static void preOrderIteration(TreeNode head) {
    if (head == null) {
        return;
    }
    Stack<TreeNode> stack = new Stack<>();
    stack.push(head);
    while (!stack.isEmpty()) {
        TreeNode node = stack.pop();
        System.out.print(node.value + " ");
        if (node.right != null) {
            stack.push(node.right);
        }
        if (node.left != null) {
            stack.push(node.left);
        }
    }
}
```
当整个左子树退栈的时候这个时候输出了该子树的根节点 2,之后输出中间节点 1。然后处理根节点为3右子树。

```java
public static void inOrderIteration(TreeNode head) {
    if (head == null) {
        return;
    }
    TreeNode cur = head;
    Stack<TreeNode> stack = new Stack<>();
    while (!stack.isEmpty() || cur != null) {
        while (cur != null) {
            stack.push(cur);
            cur = cur.left;
        }
        TreeNode node = stack.pop();
        System.out.print(node.value + " ");
        if (node.right != null) {
            cur = node.right;
        }
    }
}
```

Morris 的整体思路就是将 以某个根结点开始,找到它左子树的最右侧节点之后与这个根结点进行连接
我们可以从 图2 看到,如果这么连接之后,cur 这个指针是可以完整的从一个节点顺着下一个节点遍历,将整棵树遍历完毕,直到 7 这个节点右侧没有指向。


public static void preOrderMorris(TreeNode head) {
  if (head == null) {
      return;
  }
  TreeNode cur1 = head;//当前开始遍历的节点
  TreeNode cur2 = null;//记录当前结点的左子树
  while (cur1 != null) {
      cur2 = cur1.left;
      if (cur2 != null) {
          while (cur2.right != null && cur2.right != cur1) {//找到当前左子树的最右侧节点,且这个节点应该在指向根结点之前,否则整个节点又回到了根结点。
              cur2 = cur2.right;
          }
          if (cur2.right == null) {//这个时候如果最右侧这个节点的右指针没有指向根结点,创建连接然后往下一个左子树的根结点进行连接操作。
              cur2.right = cur1;
              cur1 = cur1.left;
              continue;
          } else {//当左子树的最右侧节点有指向根结点,此时说明我们已经回到了根结点并重复了之前的操作,同时在回到根结点的时候我们应该已经处理完 左子树的最右侧节点 了,把路断开。
              cur2.right = null;
          }
      } 
      cur1 = cur1.right;//一直往右边走,参考图
  }
}

当我们到达最左侧,也就是左边连线已经创建完毕了。
打印 4
打印 5 2
打印 6
打印 7 3 1
我们将一个节点的连续右节点当成一个单链表来看待。
当我们返回上层之后,也就是将连线断开的时候,打印下层的单链表。
比如返回到 2,此时打印 4
比如返回到 1,此时打印 5 2
比如返回到 3,此时打印 6
那么我们只需要将这个单链表逆序打印就行了,下文也给出了 单链表逆序代码
这里不应该打印当前层,而是下一层,否则根结点会先与右边打印。

public static void postOrderMorris(TreeNode head) {
  if (head == null) {
      return;
  }
  TreeNode cur1 = head;//遍历树的指针变量
  TreeNode cur2 = null;//当前子树的最右节点
  while (cur1 != null) {
      cur2 = cur1.left;
      if (cur2 != null) {
          while (cur2.right != null && cur2.right != cur1) {
              cur2 = cur2.right;
          }
          if (cur2.right == null) {
              cur2.right = cur1;
              cur1 = cur1.left;
              continue;
          } else {
              cur2.right = null;
              postMorrisPrint(cur1.left);
          }
      }
      cur1 = cur1.right;
  }
  postMorrisPrint(head);
}
//打印函数
public static void postMorrisPrint(TreeNode head) {
  TreeNode reverseList = postMorrisReverseList(head);
  TreeNode cur = reverseList;
  while (cur != null) {
      System.out.print(cur.value + " ");
      cur = cur.right;
  }
  postMorrisReverseList(reverseList);
}
//翻转单链表
public static TreeNode postMorrisReverseList(TreeNode head) {
  TreeNode cur = head;
  TreeNode pre = null;
  while (cur != null) {
      TreeNode next = cur.right;
      cur.right = pre;
      pre = cur;
      cur = next;
  }
  return pre;
}
上一篇下一篇

猜你喜欢

热点阅读