当前位置:   article > 正文

通过前序遍历序列和中序遍历序列构造二叉树_本关任务要求采用前序遍历序列和中序遍历序列构造二叉树

本关任务要求采用前序遍历序列和中序遍历序列构造二叉树

输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
在这里插入图片描述
前序遍历:根节点->左子树->右子树
中序遍历:左子树->根节点->右子树
对于前序遍历序列,可以唯一确定根节点的值,由于前序遍历是先遍历根节点,所以根节点必然是前序遍历序列的第一个元素,得到根节点后,可以在中序遍历序列中确定根节点的位置,由于中序遍历是先左子树,再根节点,然后右子树,于是就可以根据根节点的位置将左子树和右子树划分出来,然后对于每一颗左右子树都可以采用上述步骤构造,直到整颗树构造完成

class Solution {
	// 使用哈希表保存中序遍历序列中的值和序号,便于查找
    private Map<Integer, Integer> map;

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = inorder.length;
        map = new HashMap<>();
        // 将中序遍历序列的值和索引保存在哈希表中
        for (int i = 0; i < inorder.length; ++i) {
            map.put(inorder[i],i);
        }
        // 递归地构造一棵树
        return myBuidlTree(preorder, inorder, 0, n - 1, 0, n - 1);
    }

    public TreeNode myBuidlTree(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_val = preorder[preorder_left];
        // 获取根节点在中序遍历序列中的位置
        int inorder_root = map.get(preorder_root_val);
        // 构造根节点
        TreeNode root = new TreeNode(preorder_root_val);
        // 获取当前根节点左子树的大小
        int inorder_left_size = inorder_root - inorder_left;
        // 递归构造左子树
        root.left = myBuidlTree(preorder, inorder, preorder_left + 1, preorder_left + inorder_left_size, inorder_left, inorder_root - 1);
        // 递归构造右子树
        root.right = myBuidlTree(preorder, inorder, preorder_left + inorder_left_size + 1, preorder_right, inorder_root + 1, inorder_right);
        return root;
    } 
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34

同样的,根据树的后序遍历和中序遍历也可以构造二叉树,构造过程与上述过程类似,但是,根据前序遍历和后序遍历却不能构造二叉树,因为前序遍历和后序遍历不能唯一确定一棵二叉树

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/754755
推荐阅读
相关标签
  

闽ICP备14008679号