剑指 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]

方法1:把数组放到list中,方便找到索引,然后用递归。

public TreeNode buildTree(int[] preorder, int[] inorder) {
    //把前序遍历的值和中序遍历的值放到list中
    List<Integer> preorderList = new ArrayList<>();
    List<Integer> inorderList = new ArrayList<>();
    for (int i = 0; i < preorder.length; i++) {
        preorderList.add(preorder[i]);
        inorderList.add(inorder[i]);
    }
    return helper(preorderList, inorderList);
}
private TreeNode helper(List<Integer> preorderList, List<Integer> inorderList) {
    if (inorderList.size() == 0)
        return null;
    //前序遍历的第一个值就是根节点
    int rootVal = preorderList.remove(0);
    //创建跟结点
    TreeNode root = new TreeNode(rootVal);
    //查看根节点在中序遍历中的位置,然后再把中序遍历的数组劈两半,前面部分是
    //根节点左子树的所有值,后面部分是根节点右子树的所有值
    int mid = inorderList.indexOf(rootVal);
    //[0,mid)是左子树的所有值,inorderList.subList(0, mid)表示截取inorderList
    //的值,截取的范围是[0,mid),包含0不包含mid。
    root.left = helper(preorderList, inorderList.subList(0, mid));
    //[mid+1,inorderList.size())是右子树的所有值,
    // inorderList.subList(mid + 1, inorderList.size())表示截取inorderList
    //的值,截取的范围是[mid+1,inorderList.size()),包含mid+1不包含inorderList.size()。
    root.right = helper(preorderList, inorderList.subList(mid + 1, inorderList.size()));
    return root;
}

上面代码中是先把数组转化为list集合,然后在list集合中进行截取,这样效率明显不是很高,实际上我们还可以不使用list,不对数组进行截取。
这个方法不太好,主要看方法2。
方法2:

/**
 * 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) {
        return helper(0, 0, inorder.length-1, preorder, inorder);
    }

    public TreeNode helper(int preStart, int inStart, int inEnd, int[] preorder, int[] inorder){
        if(preStart>preorder.length-1||inStart>inEnd){
            return null;
        }
        TreeNode root = new TreeNode(preorder[preStart]);
        int index = 0;
        //找到当前节点root在中序遍历中的位置,然后再把数组分两半
        for(int i=inStart;i<=inEnd;i++){
            if(inorder[i]==root.val){
                index = i;
                break;
            }
        }
        root.left = helper(preStart+1, inStart, index-1, preorder, inorder);
        //这里(index-inStart+1)表示当前节点m左子树的数量加上当前节点的数量
        root.right = helper(preStart+(index-inStart+1), index+1, inEnd, preorder, inorder);
        return root;
    }
}

preStart表示的是前序遍历开始的位置
是inStart表示的是中序遍历开始的位置
inEnd表示的是中序遍历结束的位置。
思想就是找到前序数组节点在中序遍历中的位置index,然后把中序分为[0,index-1]和[index+1, inorder.length-1]。
此时前序也需要划分,左子树中前序开始的位置就是preStart+1,右子树是preStart+左子树的长度,左子树的长度是index-inStart+1,所以右子树第一个元素的所有是preStart+(index-inStart+1)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值