每天一道算法题之重建二叉树

题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。1

  • 分析: 采用递归
    • 确定父节点,前序遍历的第一个值,就是父节点值
    • 通过中序遍历,确定父节点的左子树的个数,从而可确定左子树的中序遍历和前序遍历 以及 右子树的中序遍历和前序遍历 ,调用递归即可;
    • base case:
      • 前序遍历为空时,返回null,
      • 前序遍历只有一个元素val,返回 new TreeNode(val)
  • 代码:
  	private int[] pre;
    private int[] in;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        
        pre = preorder;
        in = inorder;
        HashMap<Integer,Integer> map = new HashMap<>();//(节点值,,节点索引值)
        for(int i = 0;i < inorder.length;i++){
            map.put(inorder[i],i);
        }

        return build(0,pre.length,0,in.length,map);

    }
   
	//preS:前序遍历数组起始索引,preE:前序遍历数组长度+1
	//inS:中序遍历数组起始索引,inE:前序遍历数组长度+1
    public TreeNode build(int preS,int preE,int inS,int inE,HashMap<Integer,Integer> map){
        
        if(preS >= preE) return null;
        if(preS == preE-1) return new TreeNode(pre[preS]);

        //找父节点
        int index = map.get(pre[preS]);

        int len = index - inS;//求出左子树上的节点数,进而求出左子树的前序遍历

        TreeNode root = new TreeNode(pre[preS]);

        root.left = build(preS + 1,preS+1+len,inS,index,map);

        root.right = build(preS+1+len,preE,index+1,inE,map);

        return root;
    }

  1. https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/ ↩︎

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值