二叉树-从前序和中序中遍历序列构造二叉树

105.从前序和中序遍历序列构造二叉树

给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
提示:preorder 和 inorder 均 无重复 元素

输入:二叉树的前序和中序遍历,两个整型数组
输出:二叉树的根结点
思路:将中序遍历的结果存入哈希表中,然后通过前序遍历知道二叉树的根结点,然后递归

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        //错误输入检验
        int preLen = preorder.length;
        int inLen = inorder.length;
        if(preLen != inLen){
            throw new RuntimeException("input error");
        }
        //构建哈希表存储中序遍历的值
        Map<Integer, Integer> map = new HashMap<>();
        //值做key,下标做value
        for(int i = 0; i < inLen; i++){
            map.put(inorder[i], i);
        }
        //然后递归调用
        return myBuildTree(preorder, 0, preLen - 1, map, 0, inLen - 1);
    }
    public TreeNode myBuildTree(int[] preorder, int preLeft, int preRight, Map<Integer,Integer> map, int inLeft, int inRight){
        //递归结束条件
        if(preLeft > preRight || inLeft > inRight){
            return null;
        }
        //创建根结点
        TreeNode root = new TreeNode(preorder[preLeft]);
        int pIndex = map.get(preorder[preLeft]);
        root.left = myBuildTree(preorder, preLeft + 1, pIndex + preLeft - inLeft, map, inLeft, pIndex - 1);
        root.right = myBuildTree(preorder, pIndex + preLeft - inLeft + 1, preRight, map, pIndex + 1, inRight);
        return root;
    }
}

注意对于索引的计算
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值