温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

如何重建python二叉树

发布时间:2021-12-13 15:25:57 来源:亿速云 阅读:140 作者:柒染 栏目:大数据

如何重建python二叉树,很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

思路

前序遍历的第一个值为根节点的值,使用这个值将中序遍历结果分成两部分,左部分为树的左子树中序遍历结果,右部分为树的右子树中序遍历的结果,递归地去分别构建它的左右子树。

代码实现

package Tree;
import java.util.HashMap;
/** * 重建二叉树 * 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。 */public class Solution54 {    private HashMap<Integer, Integer> inOrderNumsIdx = new HashMap<>(); // 缓存中序遍历数组的每个值对应的索引
   public TreeNode reConstructBinaryTree(int[] pre, int[] in) {        for (int i = 0; i < in.length; i++) {            inOrderNumsIdx.put(in[i], i);        }        return reConstructBinaryTree(pre, 0, pre.length - 1, in, 0, in.length - 1);    }
   private TreeNode reConstructBinaryTree(int[] pre, int preL, int preR, int[] in, int inL, int inR) {        if (preL == preR) return new TreeNode(pre[preL]);        if (preL > preR || inL > inR) return null;        // 创建当前根节点,并为根节点赋值        TreeNode root = new TreeNode(pre[preL]);        int inIdx = inOrderNumsIdx.get(root.val);        int leftTreeSize = inIdx - inL;        // 构建左子树        root.left = reConstructBinaryTree(pre, preL + 1, preL + leftTreeSize, in, inL, inL + leftTreeSize - 1);        // 构建右子树        root.right = reConstructBinaryTree(pre, preL + leftTreeSize + 1, preR, in, inL + leftTreeSize + 1, inR);        return root;    }
   public class TreeNode {        int val;        TreeNode left;        TreeNode right;
       TreeNode(int x) {            val = x;        }    }}

看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注亿速云行业资讯频道,感谢您对亿速云的支持。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI