重建二叉树(算法题)

题目:

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

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

示例:

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

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int[] preorder;
    HashMap<Integer,Integer> in=new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        this.preorder=preorder;
        for(int i=0;i<inorder.length;i++){
            in.put(inorder[i],i);  //方便获取根节点在中序遍历的坐标
        }
        return hui(0,0,inorder.length-1);
    }
    TreeNode hui(int root,int left,int right){ //递归方法
        if(left>right) //递归结束的条件
            return null;
        TreeNode node=new TreeNode(preorder[root]);  //创建节点
        int index=in.get(preorder[root]); //获取根节点坐标
        node.left=hui(root+1,left,index-1); //获取左子树
        node.right=hui(root+index-left+1,index+1,right); 
        //root+index-left+1为当前节点+左子树个数+1
        return node;
    }
}

 总结:

        前序遍历时先是根节点,然后是左子树,右子树。

        中序遍历先是左子树,然后是根节点,最后是右子树。 

        前序遍历第一个数是二叉树最上面的根节点,遍历inorder,找到根节点的位置。根节点左边是左子树的中序遍历结果,右边是右子树的遍历结果。先找左子树,在左子树的遍历结果里最上面的一个根节点就是前序遍历的第二个数,以此类推,采用递归的方式解决问题。

        这是第一次接触到有关树的算法题,开始的时候题目都没有怎么看懂,后来看题解还是有好多地方感觉有点疑问。后来花费了好多时间才把这道题理解个差不多,记录一下,希望后来再遇到这种类型的题目时我可以自己写出来。

题目链接:https://leetcode.cn/problems/zhong-jian-er-cha-shu-lcof

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值