根据前序和中序建立二叉树

之前一直以为很简单,碰见的话肯定能写出来,结果今天面试竟然碰到了,写了半天硬是没写对,很是尴尬,
其实和我最后想的分别记录前序数组的前后位置,以及后序数组的左右位置是一样的,当时又感觉不太对,没敢写,
结果别面试跳过了。
代码:
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        return build(preorder,inorder,0,preorder.size()-1,0,inorder.size()-1);
    }
    TreeNode *build(vector<int> &preorder,vector<int> &inorder,int l1,int r1,int l2,int r2){
        if(l1>r1)
            return NULL;
        int val=preorder[l1];
        int index,cnt=0;
        for(index=l2;index<=r2&&inorder[index]!=val;index++,cnt++);
        TreeNode *tem=new TreeNode(val);
        tem->left=build(preorder,inorder,l1+1,l1+cnt,l2,index-1);
        tem->right=build(preorder,inorder,l1+cnt+1,r1,index+1,r2);
        
        return tem;
    }
};

### 如何根据前序遍历重建二叉树 #### 前提条件 为了成功地根据给定的前序(preorder)(inorder)遍历列重建一棵唯一的二叉树,必须满足以下前提条件: - 输入的两个数组均不含重复元素。 - 这两个数组代表的是同一棵二叉树的不同遍历方式。 #### 构建逻辑 前序遍历的第一个元素总是当前子树的根节点。而在中遍历的结果里找到这个根节点的位置,则可以区分出左子树右子树中的所有结点[^1]。 对于每一个子树而言,上述规则同样适用。因此可以通过递归的方法不断划分左右子树直到不能再分为止,从而完成整个二叉树结构的恢复工作。 #### Python 实现方法 下面是一个基于Python语言的具体实现方案,该函数接收两个列表作为参数——`preorder`表示前序遍历得到的节点值顺;`inorder`则对应于中遍历所获得的数据排列,并最终返回新建立好的TreeNode对象实例化后的根节点。 ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def build_tree(preorder, inorder): if not preorder or not inorder: return None root_val = preorder[0] root_index_in_inorder = inorder.index(root_val) root = TreeNode(root_val) # 左子树部分 left_preorder = preorder[1 : root_index_in_inorder + 1] left_inorder = inorder[:root_index_in_inorder] # 右子树部分 right_preorder = preorder[root_index_in_inorder + 1 :] right_inorder = inorder[root_index_in_inorder + 1:] root.left = build_tree(left_preorder, left_inorder) root.right = build_tree(right_preorder, right_inorder) return root ``` 此代码片段展示了如何利用递归来解析输入数据并逐步创建各个层次上的子树直至形成完整的二叉树结构[^2]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值