剑指offer_07_重建二叉树

最新推荐文章于 2023-07-28 21:40:10 发布

也想一个人去旅行

最新推荐文章于 2023-07-28 21:40:10 发布

阅读量128

点赞数

分类专栏：剑指offer 文章标签：二叉树

本文链接：https://blog.csdn.net/xiangfeihuanjue/article/details/104747267

版权

剑指offer 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

题目

在这里插入图片描述

分析

这是根据前序和中序遍历的结果来重建二叉树，前序：根左右。中序：左根右。那么显然前序遍历结果中的第一个元素就是二叉树的根。然后在中序遍历中找到该结点的位置，位置左边的即为左子树，右边的即为右子树，进行递归即可得到结果。
在这里插入图片描述

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder==null||preorder.length==0)
            return null;
        TreeNode treeNode=new TreeNode(preorder[0]);
        int i=0;
        for(;i<inorder.length;i++){
            if(inorder[i]==preorder[0])
                break;
        }
        if(i==inorder.length)
            throw new IllegalArgumentException("Input error");
        treeNode.left=buildTree(Arrays.copyOfRange(preorder,1,i+1),Arrays.copyOfRange(inorder,0,i));
        treeNode.right=buildTree(Arrays.copyOfRange(preorder,i+1,preorder.length),Arrays.copyOfRange(inorder,i+1,inorder.length));
        return treeNode;
    }
}