07. 重建二叉树

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

例如,给出

前序遍历 preorder = [3,9,20,15,7]  
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

方法一:递归:(图解思路请看另一篇文章)

package leecode;

import java.util.HashMap;
import java.util.Map;

/**
 * 功能描述:
 *
 * @Author: lwg
 * @Date: 2021/4/5 17:10
 */
public class demo07 {
    private Map<Integer, Integer> indexMap;

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        //获取树节点的总长度
        int n = preorder.length;
        //利用哈希映射来快速定位节点的位置
        indexMap = new HashMap<Integer, Integer>();
        for (int i = 0; i < n; i++) {
            //键代表节点的值,值代表节点的下标位置
            indexMap.put(inorder[i], i);
        }
        return myBuildTree(preorder, inorder, 0, n - 1, 0, n - 1);
    }

    public TreeNode myBuildTree(int[] preorder, int[] inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right) {
        if (preorder_left > preorder_right) {
            return null;  //此时表示遍历的位置超出
        }
        //前序遍历的第一个数就是根节点,此属性为根节点在前序遍历中的位置
        int preorder_root = preorder_left;
        //查询得到中序遍历中根节点的位置
        int inorder_root = indexMap.get(preorder[preorder_root]);
        //得到中序遍历中的左节点的个数
        int left_size = inorder_root - inorder_left;
        //构建根节点
        TreeNode root = new TreeNode(preorder[preorder_root]);
        //递归构建左子树,并连接到根节点
        root.left = myBuildTree(preorder, inorder, preorder_left + 1, preorder_left + left_size,
                inorder_left, inorder_root - 1);
        //递归构建右子树
        root.right = myBuildTree(preorder, inorder, preorder_root + left_size + 1,
                preorder_right, inorder_root + 1, inorder_right);
        return root;

    }

    class TreeNode {
        //数据域
        public int data;
        //左指针域
        public TreeNode left;
        //右指针域
        public TreeNode right;

        public TreeNode(int data, TreeNode left, TreeNode right) {
            this.data = data;
            this.left = left;
            this.right = right;
        }

        public TreeNode(int data) {
            this.data = data;
        }

        @Override
        public String toString() {
            return "leecode.demo07.TreeNode{" +
                    "data=" + data +
                    ", left=" + left +
                    ", right=" + right +
                    '}';
        }
    }

    public static void main(String[] args) {
        int preorder[] = {3, 9, 20, 15, 7};
        int inorder[] = {9, 3, 15, 20, 7};
        demo07 demo07 = new demo07();
        TreeNode treeNode = demo07.buildTree(preorder, inorder);
        System.out.println(treeNode);
    }
}

方法二: 迭代法(暂无)

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值