乐趣区

关于java:Java实现二叉树根据前序中序后序数组还原二叉树

jiezi

概述

  • 在上一篇文章中讲到顺序存储二叉树，个别是用于齐全二叉树，通过对立的数学公式能够将数组还原成齐全二叉树
  • 而对于一般的二叉树来说，也能够依据前序、中序和后序遍历失去的数组，还原二叉树

还原

  • 还原的状况分两种，别离是依据前序、中序和依据中序、后序
  • 采纳的二叉树如下图所示：
  • 留神：前序和后序是没有方法还原二叉树

依据前序、中序还原二叉树

  • 如下图所示：

  • 基本思路：

    • 依据前序数组，咱们可能确定根节点（因为前序数组的根节点肯定是第一个）
    • 中序数组依据根节点可能确定左右子树的地位。（因为中序数组的根节点肯定可能离开左右子树）
    • 前序数组依据中序数组的左右子树的地位，也可能确定本人根节点的左右子树的地位（这部分能够看看代码怎么写的）
    • 而后再别离递归左子树的前序、中序以及右子树的前序、中序，直到最初左右子树的长度都为 0
  • 代码如下：
/**
 * * @param pre 前序数组
 * @param middle 中序数组
 * @return 返回根节点
 */
public static TreeNode buildTree(int[] pre, int[] middle){if (pre == null || middle == null || pre.length == 0 || middle.length == 0 || pre.length != middle.length)
        return null;
 // 首先找到根节点
 TreeNode root = new TreeNode(pre[0]);
 // 依据根节点的值找到中序遍历中的根节点的地位索引
 int rootIndex = getIndex(middle, root.val);
 if (rootIndex == -1)
        return null;
 // 找到前序和中序中的左子树
 //copyOfRange：拷贝 [from, to) 区间里的数据，左闭右开
 int[] leftMiddle = Arrays.copyOfRange(middle, 0, rootIndex);
 int[] leftPre = Arrays.copyOfRange(pre, 1, rootIndex + 1);
 // 找到前序和中序中的右子树
 int[] rightMiddle = Arrays.copyOfRange(middle, rootIndex + 1, middle.length);
 int[] rightPre = Arrays.copyOfRange(pre, rootIndex + 1, pre.length);
 // 用递归再次构建左右子树
 TreeNode left = buildTree(leftPre, leftMiddle);
 TreeNode right = buildTree(rightPre, rightMiddle);
 // 将左右子树增加到根节点上
 root.left = left;
 root.right = right;
 return root;
}

依据中序、后序还原二叉树

  • 依据中序、后序和依据前序、中序是一样的情理，只不过是根节点从前序的第一个变为后序的最初一个，如下图所示：
  • 代码如下：
// 依据中序和后序还原二叉树
/**
 * * @param middle 中序数组
 * @param last 后序数组
 * @return 根节点
 */
public static TreeNode buildTree2(int[] middle, int[] last){if (middle == null || last == null) return null;
 int middleLen = middle.length; // 获取中序数组的长度
 int lastLen = last.length; // 获取后序数组的长度
 if (middleLen == 0 || lastLen == 0) return null;
 // 首先依据后序获取根节点
 TreeNode root = new TreeNode(last[lastLen - 1]);
 // 依据 root 获取中序的根节点的索引值
 int rootIndex = getIndex(middle, root.val);
 if (rootIndex == -1)
        return null;
 // 找到中序和后序的左子树
 int[] leftMiddle = Arrays.copyOfRange(middle, 0, rootIndex);
 int[] leftLast = Arrays.copyOfRange(last, 0, rootIndex);
 // 找到中序和后序的右子树
 int[] rightMiddle = Arrays.copyOfRange(middle, rootIndex + 1, middleLen);
 int[] rightLast = Arrays.copyOfRange(last, rootIndex, lastLen - 1);
 // 用递归再次构建左子树和右子树
 TreeNode left = buildTree2(leftMiddle, leftLast);
 TreeNode right = buildTree2(rightMiddle, rightLast);
 // 将左右子树增加到根节点
 root.left = left;
 root.right = right;
 return root;
}

总体代码

// 树节点：public class TreeNode {
    public int val;
 public TreeNode left;
 public TreeNode right;
 public TreeNode(int val){this.val = val;}
}

// 还原代码：public class Travel {public static void preorder_Traversal(TreeNode node){if (node == null)
            return;
 System.out.println(node.val);
 preorder_Traversal(node.left);
 preorder_Traversal(node.right);
 }
    public static void inorder_Traversal(TreeNode node, int val){if (node == null)
            return;
 inorder_Traversal(node.left, val);
 System.out.println("---" + node.val);
 if (node.val == val){System.out.println("找到了：" + node.val);
 return ; }
        inorder_Traversal(node.right, val);
 }
    public static void postorder_Traversal(TreeNode node){if (node == null)
            return;
 postorder_Traversal(node.left);
 postorder_Traversal(node.right);
 System.out.println(node.val);
 }
    public static int getIndex(int[] arr, int target){for (int i = 0; i < arr.length; i ++){if (arr[i] == target){return i;}
        }
        return -1;
 }
    // 依据前序和中序还原二叉树
 /**
 * * @param pre 前序数组
 * @param middle 中序数组
 * @return 返回根节点
 */
 public static TreeNode buildTree(int[] pre, int[] middle){if (pre == null || middle == null || pre.length == 0 || middle.length == 0 || pre.length != middle.length)
            return null;
 // 首先找到根节点
 TreeNode root = new TreeNode(pre[0]);
 // 依据根节点的值找到中序遍历中的根节点的地位索引
 int rootIndex = getIndex(middle, root.val);
 if (rootIndex == -1)
            return null;
 // 找到前序和中序中的左子树
 //copyOfRange：拷贝 [from, to) 区间里的数据，左闭右开
 int[] leftMiddle = Arrays.copyOfRange(middle, 0, rootIndex);
 int[] leftPre = Arrays.copyOfRange(pre, 1, rootIndex + 1);
 // 找到前序和中序中的右子树
 int[] rightMiddle = Arrays.copyOfRange(middle, rootIndex + 1, middle.length);
 int[] rightPre = Arrays.copyOfRange(pre, rootIndex + 1, pre.length);
 // 用递归再次构建左右子树
 TreeNode left = buildTree(leftPre, leftMiddle);
 TreeNode right = buildTree(rightPre, rightMiddle);
 // 将左右子树增加到根节点上
 root.left = left;
 root.right = right;
 return root;
 }
    // 依据中序和后序还原二叉树
 /**
 * * @param middle 中序数组
 * @param last 后序数组
 * @return 根节点
 */
 public static TreeNode buildTree2(int[] middle, int[] last){if (middle == null || last == null) return null;
 int middleLen = middle.length; // 获取中序数组的长度
 int lastLen = last.length; // 获取后序数组的长度
 if (middleLen == 0 || lastLen == 0) return null;
 // 首先依据后序获取根节点
 TreeNode root = new TreeNode(last[lastLen - 1]);
 // 依据 root 获取中序的根节点的索引值
 int rootIndex = getIndex(middle, root.val);
 if (rootIndex == -1)
            return null;
 // 找到中序和后序的左子树
 int[] leftMiddle = Arrays.copyOfRange(middle, 0, rootIndex);
 int[] leftLast = Arrays.copyOfRange(last, 0, rootIndex);
 // 找到中序和后序的右子树
 int[] rightMiddle = Arrays.copyOfRange(middle, rootIndex + 1, middleLen);
 int[] rightLast = Arrays.copyOfRange(last, rootIndex, lastLen - 1);
 // 用递归再次构建左子树和右子树
 TreeNode left = buildTree2(leftMiddle, leftLast);
 TreeNode right = buildTree2(rightMiddle, rightLast);
 // 将左右子树增加到根节点
 root.left = left;
 root.right = right;
 return root;
 }
    public static void main(String[] args) {
 // 前序、中序测试
//        int[] pre = {18, 20, 16, 29, 39, 8};
//        int[] middle = {16, 20, 39, 8, 29, 18};
//
//        TreeNode root = buildTree(pre, middle);
//        preorder_Traversal(root);
 // 中序、后序测试
 int[] middle = {16, 20, 18, 39, 29, 8};
 int[] last = {16, 20, 39, 8, 29, 18};
 TreeNode node = buildTree2(middle, last);
 preorder_Traversal(node);
 }
}
退出移动版