剑指 Offer 07. 重建二叉树

木琦

于 2022-08-02 10:36:18 发布

阅读量272

点赞数 1

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_38335768/article/details/126117181

版权

二叉树前序遍历中序遍历递归哈希表优化

关键词由CSDN通过智能技术生成

leetcode 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题目：剑指 Offer 07. 重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

在这里插入图片描述

示例 1:

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

示例 2:

Input: preorder = [-1], inorder = [-1]
Output: [-1]

限制：

0 <= 节点个数 <= 5000

方法一：

参考：https://leetcode.cn/problems/zhong-jian-er-cha-shu-lcof/solution/chao-ji-rong-yi-dong-de-di-gui-by-yusael-ahnb/
参考上面这位大哥的代码，简单易懂。

/**
 * 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.length==0) return null;

        int rootVal=preorder[0];

        //寻找根节点的索引
        int rootIndex=0;
        for(int i=0;i<inorder.length;i++){
            if(inorder[i]==rootVal){
                rootIndex=i;
                break;
            }
        }

        //构造根节点
        TreeNode root=new TreeNode(rootVal);

        //递归
        root.left=buildTree(
            Arrays.copyOfRange(preorder,1,rootIndex+1),
            Arrays.copyOfRange(inorder,0,rootIndex));

        root.right=buildTree(
            Arrays.copyOfRange(preorder,rootIndex+1,preorder.length),
            Arrays.copyOfRange(inorder,rootIndex+1,inorder.length));
        
        return root;

    }
}

办法笨，但有效。加油！

 Arrays.copyOfRange(preorder,1,rootIndex+1),

主要用于对一个已有的数组进行截取复制，复制出一个左闭右开区间的数组。

方法二：可用哈希表优化，以后再补充