LeetCode(105 & 106): 从前序或后序与中序遍历序列构造二叉树 Construct Binary Tree(Java)

234 篇文章 1 订阅
177 篇文章 0 订阅
这篇博客介绍了如何使用Java从先序遍历和中序遍历,以及中序遍历和后序遍历的序列来构造二叉树。博主通过改进的递归方法,结合哈希表提高了解决问题的效率。博客内容包括问题描述、解题思路以及相关注意事项。
摘要由CSDN通过智能技术生成

这道是剑指Offer和PTA原题,早年的代码是这样的:

#数据结构与算法学习笔记#剑指Offer4:先序遍历+中序遍历重建二叉树(Java、C/C++)

#数据结构与算法学习笔记#PTA11:先序遍历+中序遍历转后序遍历/二叉树非递归遍历/二叉树栈遍历(JAVA)

如今重新写一遍,追求的已经不再是简单的通过和最佳思路,而是苛求代码的简洁性和效率。相比于之前的简单递归,在其基础上又增加了哈希表用于存储先序遍历元素在中序遍历数组中的位置,大大提高了区间划分速度。后序遍历+中序遍历->构造二叉树原理相同。


传送门:从前序与中序遍历序列构造二叉树

Given preorder and inorder traversal of a tree, construct the binary tree.

Note: You may assume that duplicates do not exist in the tree.

根据一棵树的前序遍历与中序遍历构造二叉树。

注意: 你可以假设树中没有重复的元素。

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

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

传送门:从中序与后序遍历序列构造二叉树

Given inorder and postorder traversal of a tree, construct the binary tree.

Note: You may assume that duplicates do not exist in the tree.

根据一棵树的中序遍历与后序遍历构造二叉树。

注意: 你可以假设树中没有重复的元素。

例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]

返回如下的二叉树:
    3
   / \
  9  20
    /  \
   15   7



import java.util.HashMap;

/**
 *
 * Given preorder and inorder traversal of a tree, construct the binary tree.
 * You may assume that duplicates do not exist in the tree.
 * 根据一棵树的前序遍历与中序遍历构造二叉树。
 * 你可以假设树中没有重复的元素。
 *
 */
public class ConstructBinaryTreeFromPreorderAndInorderTraversal {
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

    //利用哈希表存储每一个结点的位置,再进行递归建树
    int[] preOrder;
    int[] inOrder;
    int[] postOrder;
    HashMap<Integer, Integer> map = new HashMap<>();

    //先序遍历 + 中序遍历 -> 重构二叉树
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        preOrder = preorder;
        inOrder = inOrder;
        int idx = 0;
        for(int value : inorder){
            map.put(value, idx++);
        }
        return Solution(0, 0, inorder.length - 1);
    }

    public TreeNode Solution(int rootLocPre, int begin, int end){
        if(rootLocPre >= preOrder.length || begin > end){
            return null;
        }
        TreeNode root = new TreeNode(preOrder[rootLocPre]);
        if(begin != end){
            int rootLocIn = map.get(preOrder[rootLocPre]);
            root.left = Solution(rootLocPre + 1, begin, rootLocIn - 1);
            root.right = Solution(rootLocPre + rootLocIn - begin + 1, rootLocIn + 1, end);
        }
        return root;
    }

    //后序遍历 + 中序遍历 -> 重构二叉树
    public TreeNode buildTree2(int[] inorder, int[] postorder) {
        postOrder = postorder;
        inOrder = inOrder;
        int idx = 0;
        for(int value : inorder){
            map.put(value, idx++);
        }
        return Solution2(postorder.length - 1, 0, inorder.length - 1);
    }

    public TreeNode Solution2(int rootLocPost, int begin, int end){
        if(rootLocPost < 0 || begin > end){
            return null;
        }
        TreeNode root = new TreeNode(postOrder[rootLocPost]);
        if(begin != end){
            int rootLocIn = map.get(postOrder[rootLocPost]);
            root.left = Solution2(rootLocPost - (end - rootLocIn + 1), begin, rootLocIn - 1);
            root.right = Solution2(rootLocPost - 1, rootLocIn + 1, end);
        }
        return root;
    }
}



#Coding一小时,Copying一秒钟。留个言点个赞呗,谢谢你#

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值