根据前序和中序遍历序列来构建二叉树

根据前序和中序遍历序列来构建二叉树

给定一个二叉树,我们可以写出前序遍历与中序遍历的序列,那么我们给定前序和中序遍历的序列,我们怎样才能将这颗二叉树还原呢???
给定前序遍历的数组序列为:int[] preorder,中序遍历序列数组为:int[] inorder
为了快速定位到中序遍历元素的下标,我们可以构建一个hash表来存储该元素与对应的下标。(此方法只适用于二叉树的元素的值没有相同的。)

private Map<Integer,Integer> indexMap;

  indexMap = new HashMap<>();
  for (int i = 0; i < n; i++) {
      indexMap.put(inorder[i], i);
   }

在这里插入图片描述
我们用preorder_left,preorder_right表示前序遍历的数组的第一个下标和最后一个下标,用inorder_left,inorder_right表示中序遍历数组的第一个和最后一个下标。

  • 前序遍历的第一个节点为根节点,所以前序遍历的根节点的下标:preorder_root=preorder_left。
  • 中序遍历的根节点下标我们可以从hash表中获取,所以inorder_root=indexMap.get(preorder[preorder_root])。
  • 从中序遍历的序列中,根节点左边的元素都是左子树的,所以我们可以获得左子树的节点个数size_left_subtree = inorder_root - inorder_left。
  • 获取到左子树的节点个数,我们就可以在前序遍历的序列中找到左子树最后一个节点的下标为preorder_left + size_left_subtree。

在这里插入图片描述

每个root节点的左或右子树都是一颗二叉树,所以我们可以使用递归的方式来构建。
详细的代码如下:

public class ConstructBinary {
    private Map<Integer,Integer> indexMap;

    public static void main(String[] args) {
        ConstructBinary cb = new ConstructBinary();
        int[] preorder = {3, 9, 20, 15, 7};     //前序遍历序列
        int[] inorder = {9, 3, 15, 20, 7};      //中序遍历序列
        TreeNode node = cb.buildTree(preorder, inorder);
        cb.printTree(node);

    }
    /**
     *
     * @param preorder   前序遍历的数组序列
     * @param inorder    中序遍历的数组序列
     * @return    root节点
     */
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = preorder.length;
        //构建hash表映射,便于快速定位根节点    元素是没有重复的!!!
        indexMap = new HashMap<>();
        for (int i = 0; i < n; i++) {
            indexMap.put(inorder[i], i);
        }
        return myBuildTree(preorder, inorder, 0, n - 1, 0, n - 1);
    }
    /**
     * 
     * @param preorder   前序遍历的数组序列
     * @param inorder    中序遍历的数组序列
     * @param preorder_left  前序遍历数组左边界
     * @param preorder_right  前序遍历数组右边界
     * @param inorder_left    中序遍历数组左边界
     * @param inorder_right   中序遍历数组右边界
     * @return   根节点
     */
    private TreeNode myBuildTree(int[] preorder, int[] inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right) {
        if (preorder_left > preorder_right) {
            return null;
        }
        //前序遍历的第一个节点为根节点
        int preorder_root = preorder_left;
        //从中序遍历中定位根节点
        int inorder_root = indexMap.get(preorder[preorder_root]);
        //先建立根节点
        TreeNode root = new TreeNode(preorder[preorder_root]);
        //得到左子树中节点的数目
        int size_left_subtree = inorder_root - inorder_left;
        //递归构建左子树,并连接到根节点
        root.left = myBuildTree(preorder, inorder, preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1);
        //构建右子树,并连接到根节点
        root.right = myBuildTree(preorder, inorder, preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right);
        return root;
    }

    //写个前序遍历二叉树的方法
    public void printTree(TreeNode node) {
        if (node != null) {
            System.out.println(node.val);
            //向左递归
            if (node.left != null) {
                printTree(node.left);
            }
            //向右递归
            if (node.right != null) {
                printTree(node.right);
            }
        }
    }
}
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读